[Sparse] Unified api args name (PaddlePaddle#47529)

Author: zhangkaihuo

paddle/phi/api/yaml/sparse_backward.yaml

@@ -325,7 +325,7 @@
     func : softmax_csr_grad{sparse_csr, sparse_csr -> sparse_csr}
 
 - backward_op : sparse_coo_tensor_grad
-  forward : sparse_coo_tensor(Tensor values, Tensor indices, IntArray dense_shape) -> Tensor(out)
+  forward : sparse_coo_tensor(Tensor values, Tensor indices, int64_t[] shape) -> Tensor(out)
   args : (Tensor indices, Tensor out_grad)
   output : Tensor(values_grad)
   infer_meta :

paddle/phi/api/yaml/sparse_ops.yaml

@@ -280,7 +280,7 @@
   backward : softmax_grad
 
 - op : sparse_coo_tensor
-  args : (Tensor values, Tensor indices, IntArray dense_shape)
+  args : (Tensor values, Tensor indices, int64_t[] shape={})
   output : Tensor(out)
   infer_meta :
     func : sparse::SparseCooTensorInferMeta

paddle/phi/infermeta/sparse/binary.cc

@@ -136,9 +136,9 @@ void Pool3dInferMeta(const MetaTensor& x,
 
 void SparseCooTensorInferMeta(const MetaTensor& values,
                               const MetaTensor& indices,
-                              const IntArray& dense_shape,
+                              const std::vector<int64_t>& shape,
                               MetaTensor* out) {
-  out->set_dims(phi::make_ddim(dense_shape.GetData()));
+  out->set_dims(phi::make_ddim(shape));
   out->set_dtype(values.dtype());
   out->set_layout(values.layout());
 }

paddle/phi/infermeta/sparse/binary.h

@@ -45,7 +45,7 @@ void Pool3dInferMeta(const MetaTensor& x,
 
 void SparseCooTensorInferMeta(const MetaTensor& values,
                               const MetaTensor& indices,
-                              const IntArray& dense_shape,
+                              const std::vector<int64_t>& shape,
                               MetaTensor* out);
 
 }  // namespace sparse

paddle/phi/kernels/sparse/sparse_utils_kernel.h

@@ -168,10 +168,9 @@ template <typename T, typename Context>
 void SparseCooTensorKernel(const Context& dev_ctx,
                            const DenseTensor& values,
                            const DenseTensor& indices,
-                           const IntArray& dense_shape,
+                           const std::vector<int64_t>& shape,
                            SparseCooTensor* out) {
-  *out =
-      SparseCooTensor(indices, values, phi::make_ddim(dense_shape.GetData()));
+  *out = SparseCooTensor(indices, values, phi::make_ddim(shape));
 }
 
 }  // namespace sparse

python/paddle/sparse/creation.py

@@ -169,7 +169,7 @@ def sparse_coo_tensor(
         inputs = {'values': values, 'indices': indices}
         if shape[0] is None:
             shape[0] = -1
-        attrs = {'dense_shape': shape}
+        attrs = {'shape': shape}
         helper = LayerHelper(op_type)
         out = helper.create_sparse_variable_for_type_inference(dtype)
         helper.append_op(