[PIR] Move tensor_unfold op to dygraph_ops.yaml (#69289)

gongshaotian · web-flow · commit 4d9f571ccce1 · 2024-11-12T19:16:19.000+08:00
* move tensor_unfold op to dygraph_ops.yaml

* revert conflict

* move tensor_unfold_grad
diff --git a/paddle/phi/ops/yaml/backward.yaml b/paddle/phi/ops/yaml/backward.yaml
@@ -3359,16 +3359,6 @@
     func : temporal_shift_grad
     data_type : out_grad
 
-- backward_op : tensor_unfold_grad
-  forward : tensor_unfold (Tensor input, int64_t axis, int64_t size, int64_t step) -> Tensor(out)
-  args : (Tensor input, Tensor out_grad, int64_t axis, int64_t size, int64_t step)
-  output : Tensor(input_grad)
-  infer_meta :
-    func : StridedUnChangedInferMeta
-    param : [input]
-  kernel :
-    func : tensor_unfold_grad
-
 - backward_op : thresholded_relu_grad
   forward : thresholded_relu (Tensor x, float threshold, float value) -> Tensor(out)
   args : (Tensor x, Tensor out_grad, float threshold, float value)
diff --git a/paddle/phi/ops/yaml/inconsistent/dygraph_backward.yaml b/paddle/phi/ops/yaml/inconsistent/dygraph_backward.yaml
@@ -328,6 +328,16 @@
   backward : subtract_double_grad
   inplace : (out_grad -> x_grad)
 
+- backward_op : tensor_unfold_grad
+  forward : tensor_unfold (Tensor input, int64_t axis, int64_t size, int64_t step) -> Tensor(out)
+  args : (Tensor input, Tensor out_grad, int64_t axis, int64_t size, int64_t step)
+  output : Tensor(input_grad)
+  infer_meta :
+    func : StridedUnChangedInferMeta
+    param : [input]
+  kernel :
+    func : tensor_unfold_grad
+
 - backward_op : tile_double_grad
   forward : tile_grad (Tensor x, Tensor grad_out, IntArray repeat_times) -> Tensor(grad_x)
   args : (Tensor grad_x_grad, IntArray repeat_times)
diff --git a/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml b/paddle/phi/ops/yaml/inconsistent/dygraph_ops.yaml
@@ -345,6 +345,17 @@
     data_type : DataType::FLOAT32
   traits : paddle::dialect::ForwardOnlyTrait
 
+- op : tensor_unfold
+  args : (Tensor input, int64_t axis, int64_t size, int64_t step)
+  output : Tensor
+  infer_meta :
+    func : StridedUnChangedInferMeta
+    param : [input]
+  kernel :
+    func : tensor_unfold
+  backward : tensor_unfold_grad
+  no_need_buffer : input
+
 - op : tile
   args : (Tensor x, IntArray repeat_times = {})
   output : Tensor(out)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -4949,18 +4949,6 @@
   backward : temporal_shift_grad
   interfaces : paddle::dialect::InferSymbolicShapeInterface
 
-- op : tensor_unfold
-  args : (Tensor input, int64_t axis, int64_t size, int64_t step)
-  output : Tensor
-  infer_meta :
-    func : StridedUnChangedInferMeta
-    param : [input]
-  kernel :
-    func : tensor_unfold
-  backward : tensor_unfold_grad
-  no_need_buffer : input
-  # interfaces : paddle::dialect::InferSymbolicShapeInterface
-
 - op : thresholded_relu
   args : (Tensor x, float threshold = 1.0, float value = 0.0)
   output : Tensor(out)
