| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_scatter_nd_update_op.py | 33 class ScatterNdUpdate(nn.Cell): class
35 super(ScatterNdUpdate, self).__init__()
36 self.scatter_nd_update = P.ScatterNdUpdate()
45 scatter_nd_update = ScatterNdUpdate()
56 class ScatterNdUpdate(nn.Cell): class
58 super(ScatterNdUpdate, self).__init__()
59 self.scatter_nd_update = P.ScatterNdUpdate()
68 scatter_nd_update = ScatterNdUpdate()
79 class ScatterNdUpdate(nn.Cell): class
81 super(ScatterNdUpdate, self).__init__()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | scatter_nd_functor_gpu_kernel.cc | 22 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
29 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
36 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
43 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
50 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
57 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
64 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
71 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
78 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
85 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate,
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | scatter_nd_update.h | 31 class MS_CORE_API ScatterNdUpdate : public PrimitiveC {
34 …ScatterNdUpdate() : PrimitiveC(kNameScatterNdUpdate) { InitIOName({"input_x", "indices", "update"}… in ScatterNdUpdate() function
36 ~ScatterNdUpdate() = default;
37 MS_DECLARE_PARENT(ScatterNdUpdate, PrimitiveC);
43 using PrimScatterNdUpdatePtr = std::shared_ptr<ScatterNdUpdate>;
|
| D | scatter_nd_update.cc | 56 REGISTER_PRIMITIVE_C(kNameScatterNdUpdate, ScatterNdUpdate);
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | matrix_calculation_ops_declare.cc | 33 INPUT_MAP(ScatterNdUpdate) = {{1, INPUT_DESC(var)}, {2, INPUT_DESC(indices)}, {3, INPUT_DESC(update…
34 ATTR_MAP(ScatterNdUpdate) = {{"use_locking", ATTR_DESC(use_locking, AnyTraits<bool>())}};
35 OUTPUT_MAP(ScatterNdUpdate) = {{0, OUTPUT_DESC(var)}};
36 REG_ADPT_DESC(ScatterNdUpdate, kNameScatterNdUpdate, ADPT_DESC(ScatterNdUpdate))
|
| D | matrix_calculation_ops_declare.h | 32 DECLARE_OP_ADAPTER(ScatterNdUpdate)
33 DECLARE_OP_USE_OUTPUT(ScatterNdUpdate)
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_scatter_nd_parser.cc | 25 auto prim = std::make_unique<ops::ScatterNdUpdate>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scatter_nd_update_fp32.cc | 111 int ScatterNdUpdateCPUKernel::ScatterNdUpdate(int task_id) { in ScatterNdUpdate() function in mindspore::kernel::ScatterNdUpdateCPUKernel
130 auto ret = g_kernel->ScatterNdUpdate(task_id); in ScatterNdUpdateRun()
|
| D | scatter_nd_update_fp32.h | 35 int ScatterNdUpdate(int task_id);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | scatter_nd_update_infer.c | 37 REG_INFER(ScatterNdUpdate, PrimType_ScatterNdUpdate, ScatterNdUpdateInferShape)
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_while_ScatterNdUpdate.py | 29 self.upd = P.ScatterNdUpdate()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | scatter_nd_update_cpu_kernel.h | 60 MS_REG_CPU_KERNEL(ScatterNdUpdate,
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_signature.py | 61 self.scatter = P.ScatterNdUpdate()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | scatter_nd_functor_impl.cu | 21 __global__ void ScatterNdUpdate(const size_t unit_size, const size_t index_depth, const size_t upda… in ScatterNdUpdate() function
105 return ScatterNdUpdate<<<GET_BLOCKS(updates_size), GET_THREADS, 0, cuda_stream>>>( in CalScatterNdFunctor()
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 465 FUNC_MSOP2SCHEMAOP_DECLARE(ScatterNdUpdate)
|
| D | ops_def.cc | 190 OP_TYPE(ScatterNdUpdate)
1203 OP_SCHEMA_DEF(ScatterNdUpdate)
1204 OP_SCHEMA_DEF_END(ScatterNdUpdate)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 29 … ScatterUpdate, ScalarToArray, ScalarToTensor, ScatterNd, ScatterNdUpdate, Select,
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_scatter_nd_func_op.py | 28 "update": ops.ScatterNdUpdate,
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops.fbs | 191 ScatterNdUpdate,
1203 table ScatterNdUpdate {
|
| D | model_generated.h | 977 const mindspore::schema::ScatterNdUpdate *value_as_ScatterNdUpdate() const { in value_as_ScatterNdUpdate()
978 …PrimitiveType_ScatterNdUpdate ? static_cast<const mindspore::schema::ScatterNdUpdate *>(value()) :… in value_as_ScatterNdUpdate()
1756 …mplate<> inline const mindspore::schema::ScatterNdUpdate *Primitive::value_as<mindspore::schema::S…
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
| D | bbox_assign_sample.py | 84 self.scatterNdUpdate = P.ScatterNdUpdate()
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 140 scatter_nd_update = P.ScatterNdUpdate()
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_ops.py | 309 self.scatter_nd_update = P.ScatterNdUpdate()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 746 @bprop_getters.register(P.ScatterNdUpdate)
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | model_generated.h | 1126 const mindspore::schema::ScatterNdUpdate *value_as_ScatterNdUpdate() const { in value_as_ScatterNdUpdate()
1127 …PrimitiveType_ScatterNdUpdate ? static_cast<const mindspore::schema::ScatterNdUpdate *>(value()) :… in value_as_ScatterNdUpdate()
1911 …mplate<> inline const mindspore::schema::ScatterNdUpdate *Primitive::value_as<mindspore::schema::S…
|