Home
last modified time | relevance | path

Searched defs:TensorScatterAdd (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dtensor_scatter_add.h34 …TensorScatterAdd() : BaseOperator(kNameTensorScatterAdd) { InitIOName({"input_x", "indices", "upda… in TensorScatterAdd() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dtensor_scatter_add.cc38 int TensorScatterAddCPUKernel::TensorScatterAdd(int task_id) { in TensorScatterAdd() function in mindspore::kernel::TensorScatterAddCPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/base/
Dmeta_op.h58 TensorScatterAdd, enumerator
Dir_builder.h193 …inline NodePtr TensorScatterAdd(const NodePtr &input, const NodePtr &indices, const NodePtr &updat… in TensorScatterAdd() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dgraph_builder.h67 …NodePtr TensorScatterAdd(const NodePtr &input, const NodePtr &indices, const NodePtr &update) cons… in TensorScatterAdd() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Darray_ops.py4667 class TensorScatterAdd(Primitive): class