Searched refs:scatter_nd_update (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_scatter_nd_update_op.py | 36 self.scatter_nd_update = P.ScatterNdUpdate()
40 return self.scatter_nd_update(self.x, indices, update)
45 scatter_nd_update = ScatterNdUpdate()
46 scatter_nd_update(indices, update)
47 print("x:\n", scatter_nd_update.x.data.asnumpy())
49 assert np.allclose(scatter_nd_update.x.data.asnumpy(), np.array(expect, np.float))
59 self.scatter_nd_update = P.ScatterNdUpdate()
63 return self.scatter_nd_update(self.x, indices, update)
68 scatter_nd_update = ScatterNdUpdate()
69 scatter_nd_update(indices, update)
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | tensor_scatter_update_fission.cc | 45 auto scatter_nd_update = graph->NewCNode(inputs); in CreateScatterNdUpdate() local
46 MS_EXCEPTION_IF_NULL(scatter_nd_update); in CreateScatterNdUpdate()
47 scatter_nd_update->set_scope(tensor_scatter_update->scope()); in CreateScatterNdUpdate()
48 scatter_nd_update->set_abstract(tensor_scatter_update->abstract()); in CreateScatterNdUpdate()
49 return scatter_nd_update; in CreateScatterNdUpdate()
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | tensor_scatter_update_fission_test.py | 21 scatter_nd_update = Primitive('ScatterNdUpdate') variable
48 res = scatter_nd_update(res, indices, updates)
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_ops.py | 309 self.scatter_nd_update = P.ScatterNdUpdate()
312 self.scatter_nd_update(self.input_x, indices, updates)
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 140 scatter_nd_update = P.ScatterNdUpdate() variable
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 248 from .scatter_nd_update import _scatter_nd_update_tbe
|
| /third_party/mindspore/config/ |
| D | op_info.config | 367 …"async_flag": false, "binfile_name": "scatter_nd_update.so", "compute_cost": 10, "kernel_name": "s…
|