diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py index a5ca03b2bc966988faf6e09f874200702c3f7510..89530cd4fb2e8a9305ea485cfbddd671020bd3ac 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(