mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

update akg/modify comment

This commit is contained in:
Yang Jiao 2021-12-02 15:43:34 +08:00
parent 22f4909784
commit 0f354754d8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
akg

@ -1 +1 @@
Subproject commit b3f67ab15c03ab7b201152d41cdfad7c05294609
Subproject commit 3a7415a9a4184e3d8469b1d19109c03a32843871

2
mindspore/ops/operations/custom_ops.py
View File

@ -76,7 +76,7 @@ class Custom(ops.PrimitiveWithInfer):
then nparam=5
- params(void \*\*): a pointer to the array of inputs and outputs' pointer; the pointer type of inputs
and outputs is void \* ; suppose the operator has 2 inputs and 3 outputs, then the first input's
pointer is nparam[0] and the second output's pointer is nparam[4]
pointer is params[0] and the second output's pointer is params[3]
- ndims(int \*): a pointer to the array of inputs and outputs' dimension num; suppose params[i] is a
1024x1024 tensor and params[j] is a 77x83x4 tensor, then ndims[i]=2, ndims[j]=3.
- shapes(int64_t \*\*): a pointer to the array of inputs and outputs' shapes(int64_t \*); the ith