From 3cf1080fb9837ab781dec4ce12f0ae8bae3e58fc Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 19 Sep 2019 16:41:31 +0200
Subject: [PATCH] Fix naming of tensorflow wrapping functions
---
src/pystencils_autodiff/backends/_tensorflow.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py
index a5ca03b..89530cd 100644
--- a/src/pystencils_autodiff/backends/_tensorflow.py
+++ b/src/pystencils_autodiff/backends/_tensorflow.py
@@ -51,10 +51,10 @@ 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}_{_hash(str(autodiff_obj).encode()).hexdigest()}'
+ autodiff_obj.op_name = f'{autodiff_obj.op_name}_{_hash(str(autodiff_obj).encode()).hexdigest()}'
forward_ast.function_name = autodiff_obj.op_name + "_forward"
backward_ast.function_name = autodiff_obj.op_name + "_backward"
- module = TensorflowModule(op_name, [forward_ast, backward_ast])
+ module = TensorflowModule(autodiff_obj.op_name, [forward_ast, backward_ast])
compiled_op = module.compile()
backward_func = getattr(compiled_op, stringcase.snakecase(
--
GitLab