Browse Source

update akg/modify comment

tags/v1.6.0
Yang Jiao 4 years ago
parent
commit
0f354754d8
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      akg
  2. +1
    -1
      mindspore/ops/operations/custom_ops.py

+ 1
- 1
akg

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

+ 1
- 1
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


Loading…
Cancel
Save