Searched refs:ScatterNonAliasingAdd (Results 1 – 8 of 8) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
D | selection_ops_declare.cc | 91 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))
|
D | selection_ops_declare.h | 34 DECLARE_OP_ADAPTER(ScatterNonAliasingAdd) 35 DECLARE_OP_USE_OUTPUT(ScatterNonAliasingAdd)
|
/third_party/mindspore/mindspore/ops/operations/ |
D | __init__.py | 31 … UniqueWithPad, ScatterNdAdd, ScatterNdSub, ScatterNonAliasingAdd, ReverseV2, Rint,
|
D | array_ops.py | 4750 class ScatterNonAliasingAdd(_ScatterNdOp): class
|
/third_party/mindspore/tests/st/auto_monad/ |
D | test_effect_ops.py | 334 self.scatter_non_aliasing_add = P.ScatterNonAliasingAdd()
|
/third_party/mindspore/tests/ut/python/ops/ |
D | test_ops.py | 465 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/ |
D | grad_array_ops.py | 757 @bprop_getters.register(P.ScatterNonAliasingAdd)
|
/third_party/mindspore/config/ |
D | op_info.config | 237 {"op_name": "ScatterNonAliasingAdd", "inputs": [{"index": 0, "name": "x", "need_compile": false, "p…
|