Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
pycodegen
pystencils_autodiff
Commits
253c5a9c
Commit
253c5a9c
authored
Dec 16, 2019
by
Stephan Seitz
Browse files
💄
Lint
parent
52564579
Pipeline
#20552
passed with stage
in 4 minutes and 46 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
src/pystencils_autodiff/backends/_tensorflow.py
View file @
253c5a9c
...
...
@@ -51,7 +51,7 @@ def native_tensorflowop_from_autodiffop(autodiff_obj: pystencils_autodiff.AutoDi
forward_ast
=
autodiff_obj
.
forward_ast_cpu
backward_ast
=
autodiff_obj
.
backward_ast_cpu
op_name
=
f
'
{
autodiff_obj
.
op_name
.
lower
()
}
_hash
{
_hash
(
str
((
autodiff_obj
,
autodiff_obj
.
forward_input_fields
,
autodiff_obj
.
backward_input_fields
)).
encode
()).
hexdigest
()
}
'
op_name
=
f
'
{
autodiff_obj
.
op_name
.
lower
()
}
_hash
{
_hash
(
str
((
autodiff_obj
,
autodiff_obj
.
forward_input_fields
,
autodiff_obj
.
backward_input_fields
)).
encode
()).
hexdigest
()
}
'
# noqa
if
use_cuda
:
op_name
+=
'_cuda'
forward_ast
.
function_name
=
op_name
+
"_forward"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment