Home
last modified time | relevance | path

Searched refs:ScatterNonAliasingAdd (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dselection_ops_declare.cc91 INPUT_MAP(ScatterNonAliasingAdd) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(indices)}, {3, INPUT_DESC(up…
92 ATTR_MAP(ScatterNonAliasingAdd) = EMPTY_ATTR_MAP;
93 OUTPUT_MAP(ScatterNonAliasingAdd) = {{0, OUTPUT_DESC(y)}};
94 REG_ADPT_DESC(ScatterNonAliasingAdd, kNameScatterNonAliasingAdd, ADPT_DESC(ScatterNonAliasingAdd))
Dselection_ops_declare.h34 DECLARE_OP_ADAPTER(ScatterNonAliasingAdd)
35 DECLARE_OP_USE_OUTPUT(ScatterNonAliasingAdd)
/third_party/mindspore/mindspore/ops/operations/
D__init__.py31 … UniqueWithPad, ScatterNdAdd, ScatterNdSub, ScatterNonAliasingAdd, ReverseV2, Rint,
Darray_ops.py4750 class ScatterNonAliasingAdd(_ScatterNdOp): class
/third_party/mindspore/tests/st/auto_monad/
Dtest_effect_ops.py334 self.scatter_non_aliasing_add = P.ScatterNonAliasingAdd()
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py465 class ScatterNonAliasingAdd(nn.Cell): class
469 super(ScatterNonAliasingAdd, self).__init__()
470 self.scatter_no_aliasing_add = P.ScatterNonAliasingAdd()
2715 'block': ScatterNonAliasingAdd((8,)),
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py757 @bprop_getters.register(P.ScatterNonAliasingAdd)
/third_party/mindspore/config/
Dop_info.config237 {"op_name": "ScatterNonAliasingAdd", "inputs": [{"index": 0, "name": "x", "need_compile": false, "p…