Searched refs:scatter_update (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_scatter_update.py | 31 self.scatter_update = P.ScatterUpdate().shard(strategy1)
36 out = self.scatter_update(self.inputs, self.indices, self.updates)
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | lazyadam.py | 56 scatter_update = P.ScatterUpdate(use_locking)
73 success = F.depend(success, scatter_update(m, indices, next_m))
74 success = F.depend(success, scatter_update(v, indices, next_v))
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_ops.py | 234 self.scatter_update = P.ScatterUpdate()
237 self.scatter_update(self.input_x, indices, updates)
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 138 scatter_update = P.ScatterUpdate() variable
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_scatter_arithmetic_op.py | 583 self.scatter_update = P.ScatterUpdate(use_locking=lock)
589 out = self.scatter_update(self.inputx, self.indices, self.updates)
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 284 from .scatter_update import _scatter_update_tbe
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 418 self.scatter_update = P.ScatterUpdate(use_locking)
422 out = self.scatter_update(self.ref, indices, updates)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 403 …", "async_flag": false, "binfile_name": "scatter_update.so", "compute_cost": 10, "kernel_name": "s…
404 …", "async_flag": false, "binfile_name": "scatter_update.so", "compute_cost": 10, "kernel_name": "s…
471 …", "async_flag": false, "binfile_name": "scatter_update.so", "compute_cost": 10, "kernel_name": "s…
|