Skip to content
Snippets Groups Projects
Commit 983dec1b authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fixup: add ASTs to AutoDiffOp

parent 34c303c5
No related branches found
No related tags found
No related merge requests found
Pipeline #18046 passed
...@@ -62,6 +62,10 @@ Backward: ...@@ -62,6 +62,10 @@ Backward:
self._time_constant_fields = time_constant_fields self._time_constant_fields = time_constant_fields
self._kwargs = kwargs self._kwargs = kwargs
self.op_name = op_name self.op_name = op_name
self._forward_ast_cpu = None
self._backward_ast_cpu = None
self._forward_ast_gpu = None
self._backward_ast_gpu = None
self._forward_kernel_cpu = None self._forward_kernel_cpu = None
self._backward_kernel_cpu = None self._backward_kernel_cpu = None
self._forward_kernel_gpu = None self._forward_kernel_gpu = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment