From 0f354754d8445d73186d4a1e7e7990e2b3ddb097 Mon Sep 17 00:00:00 2001 From: Yang Jiao Date: Thu, 2 Dec 2021 15:43:34 +0800 Subject: [PATCH] update akg/modify comment --- akg | 2 +- mindspore/ops/operations/custom_ops.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/akg b/akg index b3f67ab15c0..3a7415a9a41 160000 --- a/akg +++ b/akg @@ -1 +1 @@ -Subproject commit b3f67ab15c03ab7b201152d41cdfad7c05294609 +Subproject commit 3a7415a9a4184e3d8469b1d19109c03a32843871 diff --git a/mindspore/ops/operations/custom_ops.py b/mindspore/ops/operations/custom_ops.py index 3e1159d271b..0e6c2853d5c 100644 --- a/mindspore/ops/operations/custom_ops.py +++ b/mindspore/ops/operations/custom_ops.py @@ -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