Home
last modified time | relevance | path

Searched defs:TensorScatterUpdate (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dtensor_scatter_update.h34 TensorScatterUpdate() : BaseOperator(kNameTensorScatterUpdate) { in TensorScatterUpdate() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dmatrix_calculation_ops_proto.cc185 IMPLEMT_VERIFIER(TensorScatterUpdate, TensorScatterUpdateVerify) { in IMPLEMT_VERIFIER() argument
197 IMPLEMT_INFERFUNC(TensorScatterUpdate, TensorScatterUpdateInferShape) { in IMPLEMT_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/
Demitter.h164 …NodePtr TensorScatterUpdate(const NodePtr &input_x, const NodePtr &indices, const NodePtr &updates… in TensorScatterUpdate() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Darray_ops.py4451 class TensorScatterUpdate(_TensorScatterOp): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini5190 [TensorScatterUpdate] section
5191 opInfo.subTypeOfInferShape = 1
5192 opInfo.opsFlag = OPS_FLAG_CLOSE
5193 opInfo.engine = DNN_VM_AICPU
5194 opInfo.flagPartial = False
5195 opInfo.computeCost = 100
5196 opInfo.flagAsync = False
5197 opInfo.opKernelLib = CUSTAICPUKernel
5198 opInfo.formatAgnostic = False
5199 opInfo.userDefined = True
[all …]