fix

PaddlePaddle · Aug 24, 2023 · 52a4185 · 52a4185
1 parent 33df9c9
commit 52a4185
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/ir/dialect/paddle_dialect/ir/pd_op.yaml b/paddle/fluid/ir/dialect/paddle_dialect/ir/pd_op.yaml
@@ -376,7 +376,7 @@
     func: [set_value]
     param: [x, starts, ends, steps, axes, decrease_axes, none_axes, shape, values]
   inplace: null
-  backward: null
+  backward: set_value_grad
 
 - name : set_value_with_tensor
   inputs:
@@ -398,7 +398,7 @@
     func: [set_value_with_tensor]
     param: [x, values, starts, ends, steps, axes, decrease_axes, none_axes]
   inplace: null
-  backward: null
+  backward: set_value_grad
 
 
 - name : set_value_grad