[CINN]Add InferSymbolicShape for tensor_to_array (#69069)

* test

* Refine logic

* Refine logic

* revert useless code

* apply review

* fix codestyle

* revert select_output

* fix codestyle

Author: ooooo-create

paddle/fluid/pir/dialect/operator/ir/manual_op.cc

@@ -2613,6 +2613,17 @@ std::vector<pir::Type> TensorToArrayOp::InferMeta(
   return argument_outputs;
 }
 
+bool TensorToArrayOp::InferSymbolicShape(
+    pir::InferSymbolicShapeContext *infer_context) {
+  const auto &x_shape_or_data =
+      infer_context->GetShapeOrDataForValue(x())
+          .dyn_cast<symbol::RankedTensorArrayShapeOrDataDimExprs>();
+  infer_context->SetShapeOrDataForValue(
+      x_grad(), symbol::ShapeOrDataDimExprs{x_shape_or_data});
+
+  return true;
+}
+
 OpInfoTuple SliceArrayOp::GetOpInfo() {
   std::vector<paddle::dialect::OpInputInfo> inputs = {
       paddle::dialect::OpInputInfo("input",

paddle/fluid/pir/dialect/operator/ir/manual_op.h

@@ -407,8 +407,10 @@ class TEST_API ArrayToTensorOp : public pir::Op<ArrayToTensorOp,
       const std::vector<std::vector<bool>> &stop_gradients);
 };
 
-class TEST_API TensorToArrayOp
-    : public pir::Op<TensorToArrayOp, OpYamlInfoInterface, InferMetaInterface> {
+class TEST_API TensorToArrayOp : public pir::Op<TensorToArrayOp,
+                                                OpYamlInfoInterface,
+                                                InferMetaInterface,
+                                                InferSymbolicShapeInterface> {
  public:
   using Op::Op;
   static const char *name() { return "pd_op.tensor_to_array"; }
@@ -429,6 +431,7 @@ class TEST_API TensorToArrayOp
   static std::vector<pir::Type> InferMeta(
       const std::vector<pir::Value> &input_values,
       pir::AttributeMap *p_attributes);
+  bool InferSymbolicShape(pir::InferSymbolicShapeContext *infer_context);
 };
 
 class TEST_API SliceArrayOp