Searched refs:_get_group (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 21 from ...communication.management import get_rank, get_group_size, GlobalComm, _get_group
151 if not isinstance(_get_group(group), str):
156 self.add_prim_attr('group', _get_group(group))
225 validator.check_value_type('group', _get_group(group), (str,), self.name)
226 self.rank = get_rank(_get_group(group))
227 self.rank_size = get_group_size(_get_group(group))
230 self.add_prim_attr('group', _get_group(group))
262 validator.check_value_type('group', _get_group(group), (str,), self.name)
263 self.rank = get_rank(_get_group(group))
264 self.rank_size = get_group_size(_get_group(group))
[all …]
|
| /third_party/mindspore/mindspore/communication/ |
| D | management.py | 33 def _get_group(group): function
193 return _get_rank_helper(group=_get_group(group), backend=GlobalComm.BACKEND)
227 return _get_local_rank_helper(group=_get_group(group), backend=GlobalComm.BACKEND)
260 return _get_size_helper(group=_get_group(group), backend=GlobalComm.BACKEND)
293 return _get_local_size_helper(group=_get_group(group), backend=GlobalComm.BACKEND)
|