diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py index 723749c5c1f52ae0333177e529ab735821ad0174..8c78c84a466894863d5fe814f9524d2290681e9e 100644 --- a/src/pystencils_autodiff/_autodiff.py +++ b/src/pystencils_autodiff/_autodiff.py @@ -35,7 +35,7 @@ Backward: def __init__(self, forward_assignments: List[ps.Assignment], - op_name: str = "AutoDiffOp", + op_name: str = "autodiffop", time_constant_fields: List[ps.Field] = [], constant_fields: List[ps.Field] = [], diff_fields_prefix='diff', # TODO: remove! diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py index db85df108fcc0907aea70e59e134046cc023dad3..59a66f48c7f78f04ecccc4d3226a84d5b726fb9a 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 - autodiff_obj.op_name = f'{autodiff_obj.op_name}_{_hash(str(autodiff_obj).encode()).hexdigest()}' + autodiff_obj.op_name = f'{autodiff_obj.op_name}_hash{_hash(str(autodiff_obj).encode()).hexdigest()}' if use_cuda: autodiff_obj.op_name += '_cuda' forward_ast.function_name = autodiff_obj.op_name + "_forward"