feat(pir): reg pull_sparse_v2

PaddlePaddle · Apr 7, 2024 · 015a785 · 015a785
1 parent 5be2837
commit 015a785
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/phi/api/yaml/backward.yaml b/paddle/phi/api/yaml/backward.yaml
@@ -1837,9 +1837,9 @@
 
 - backward_op : pull_sparse_v2_grad
   forward : pull_sparse_v2 (Tensor[] ids, Tensor[] w, int embeddingdim = 11, int tableid = 0, str accessorclass = "", str ctrlabelname = "", int paddingid = 0, bool scalesparsegrad = true, str[] inputnames = {}, bool is_distributed = true) -> Tensor[](out)
-  args : (Tensor[] ids, Tensor[] w, Tensor[] out_grad_in, int embeddingdim, int tableid, str accessorclass, str ctrlabelname, int paddingid, bool scalesparsegrad, str[] inputnames, bool is_distributed)
+  args : (Tensor[] ids, Tensor[] w, Tensor[] out_grad, int embeddingdim, int tableid, str accessorclass, str ctrlabelname, int paddingid, bool scalesparsegrad, str[] inputnames, bool is_distributed)
   output : Tensor[](out_grad_out)
-  invoke : push_sparse_v2(ids, w, out_grad_in, embeddingdim, tableid, accessorclass, ctrlabelname, paddingid, scalesparsegrad, inputnames, is_distributed)
+  invoke : push_sparse_v2(ids, w, out_grad, embeddingdim, tableid, accessorclass, ctrlabelname, paddingid, scalesparsegrad, inputnames, is_distributed)
 
 - backward_op : put_along_axis_grad
   forward : put_along_axis (Tensor arr, Tensor indices, Tensor values, int axis, str reduce = "assign", bool include_self = true) -> Tensor(out)