Searched defs:split_indices (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | thor.py | 252 …use_nesterov=False, decay_filter=lambda x: x.name not in [], split_indices=None, enable_clip_grad=… argument
374 use_nesterov=False, decay_filter=lambda x: x.name not in [], split_indices=None, argument
444 def _define_gpu_reducer(self, split_indices): argument
665 …decay_filter=lambda x: x.name not in [], split_indices=None, enable_clip_grad=False, frequency=100… argument
763 def _define_ascend_reducer(self, split_indices): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/ |
| D | grad_reducer.py | 57 def _init_allreduce_operators(length, split_indices, group=GlobalComm.WORLD_COMM_GROUP): argument
85 def _init_allreduce_operators_by_parameters(parameters, split_indices, group, fusion_type=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | communication_op_fusion.cc | 261 std::vector<uint32_t> split_indices; in GetSplitSegments() local
|