Searched defs:split_indices (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | grad_reducer.py | 30 def _init_allreduce_operators(length, split_indices, group=GlobalComm.WORLD_COMM_GROUP): argument
52 def _init_allreduce_operators_by_parameters(parameters, split_indices, group, fusion_type=1): argument
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 245 …use_nesterov=False, decay_filter=lambda x: x.name not in [], split_indices=None, enable_clip_grad=… argument
373 use_nesterov=False, decay_filter=lambda x: x.name not in [], split_indices=None, argument
444 def _define_gpu_reducer(self, split_indices): argument
664 …decay_filter=lambda x: x.name not in [], split_indices=None, enable_clip_grad=False, frequency=100… argument
762 def _define_ascend_reducer(self, split_indices): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 148 use_nesterov=False, decay_filter=lambda x: x.name not in [], split_indices=None): argument
160 decay_filter=lambda x: x.name not in [], split_indices=None): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | communication_op_fusion.cc | 157 std::vector<uint32_t> split_indices; in GetSplitSegments() local
|