Searched refs:_get_device_num (Results 1 – 17 of 17) sorted by relevance
| /third_party/mindspore/mindspore/train/ |
| D | dataset_helper.py | 23 from ..parallel._utils import _get_device_num, _get_global_rank, _need_to_full, _to_full_shapes, _g…
163 device_num = _get_device_num() // _get_pipeline_stages()
414 batch_expand_num = _get_device_num() // _get_pipeline_stages()
449 device_num = _get_device_num() // _get_pipeline_stages()
494 self.device_num = _get_device_num()
|
| D | model.py | 28 from ..parallel._utils import _get_parallel_mode, _get_device_num, _get_global_rank, \
148 self._device_number = _get_device_num()
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | dataset_helper.py | 17 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _to_full_shapes
126 device_num = _get_device_num()
|
| D | model_thor.py | 30 from mindspore.parallel._utils import _get_parallel_mode, _get_device_num, _get_global_rank, \
150 self._device_number = _get_device_num()
|
| D | thor.py | 24 from mindspore.parallel._utils import _get_device_num, _get_gradients_mean
239 degree = _get_device_num()
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | cell_wrapper.py | 17 from mindspore.parallel._utils import (_get_device_num, _get_gradients_mean,
94 degree = _get_device_num()
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | dataset_helper.py | 22 from mindspore.parallel._utils import _get_device_num, _need_to_full, _to_full_shapes
170 device_num = _get_device_num()
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 177 def _get_device_num(): function
317 local_rank = int(_get_global_rank() % (_get_device_num() / _get_pipeline_stages()))
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | cell_wrapper.py | 18 from mindspore.parallel._utils import (_get_device_num, _get_gradients_mean,
349 self.degree = _get_device_num()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 25 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean
298 degree = _get_device_num()
|
| /third_party/mindspore/mindspore/boost/ |
| D | boost_cell_wrapper.py | 19 from mindspore.parallel._utils import _get_global_rank, _get_device_num, _get_gradients_mean
160 self.degree = _get_device_num()
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 30 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean
313 degree = _get_device_num()
|
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 32 from ..parallel._utils import _get_device_num, _get_global_rank, _need_to_full, _check_full_batch, …
542 args_full = _to_full_tensor(args, _get_device_num(), _get_global_rank())
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | optimizer.py | 32 from mindspore.parallel._utils import _get_global_rank, _get_device_num, _get_parallel_mode
223 self.dev_num = _get_device_num()
|
| D | thor.py | 26 from mindspore.parallel._utils import _get_device_num, _get_gradients_mean
450 degree = _get_device_num()
768 degree = _get_device_num()
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 1451 from mindspore.parallel._utils import _get_global_rank, _get_device_num
1453 device_num = _get_device_num()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 53 from mindspore.parallel._utils import _get_device_num
3724 if platform.system() != "Windows" and _get_device_num() >= 1:
3725 shm_estimate_usage = _get_device_num() * num_worker * num_queues * \
|