diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py index 5286743989f1700c7c9879dde98dad2642b98d89..decaf32ab6414cb5fdf8b7a899db8e22b9a9c5af 100644 --- a/src/pystencils_autodiff/backends/_tensorflow.py +++ b/src/pystencils_autodiff/backends/_tensorflow.py @@ -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).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()}' if use_cuda: op_name += '_cuda' forward_ast.function_name = op_name + "_forward"