Searched refs:set_all_reduce_fusion_split_indices (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/parallel/ |
| D | _auto_parallel_context.py | 350 def set_all_reduce_fusion_split_indices(self, indices, group=""): member in _AutoParallelContext
381 self._context_handle.set_all_reduce_fusion_split_indices(indices, new_group)
648 "all_reduce_fusion_config": auto_parallel_context().set_all_reduce_fusion_split_indices,
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 241 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
242 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
243 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
244 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 455 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
456 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
774 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
775 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
779 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
780 …auto_parallel_context().set_all_reduce_fusion_split_indices(self.split_indices, "hccl_world_groups…
|