Searched refs:_get_parallel_mode (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 29 def _get_parallel_mode(): function
35 return _get_parallel_mode() in [ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL]
68 parallel_mode = _get_parallel_mode()
76 if _get_parallel_mode() not in ("semi_auto_parallel", "auto_parallel"):
276 if _get_parallel_mode() in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | op_parallel_config.py | 23 from mindspore.parallel._utils import _get_parallel_mode
139 …is_auto_parallel = _get_parallel_mode() in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PAR…
|
| D | transformer.py | 32 from mindspore.parallel._utils import _get_parallel_mode
698 …self.is_parallel_mode = _get_parallel_mode() in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUT…
1894 if _get_parallel_mode() in (ParallelMode.AUTO_PARALLEL,):
2085 if _get_parallel_mode() in (ParallelMode.AUTO_PARALLEL,):
2310 if _get_parallel_mode() in (ParallelMode.AUTO_PARALLEL,):
|
| /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
125 if _get_parallel_mode() in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
|
| D | model_thor.py | 30 from mindspore.parallel._utils import _get_parallel_mode, _get_device_num, _get_global_rank, \
149 self._parallel_mode = _get_parallel_mode()
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | cell_wrapper.py | 18 _get_parallel_mode)
89 self.parallel_mode = _get_parallel_mode()
|
| /third_party/mindspore/mindspore/train/ |
| D | amp.py | 23 from ..parallel._utils import _get_parallel_mode, _get_pipeline_stages
189 if _get_parallel_mode() in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
|
| D | model.py | 28 from ..parallel._utils import _get_parallel_mode, _get_device_num, _get_global_rank, \
147 self._parallel_mode = _get_parallel_mode()
952 … if _get_parallel_mode() not in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
954 "mode, but got {}.".format(_get_parallel_mode()))
1072 … if _get_parallel_mode() not in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | optimizer.py | 32 from mindspore.parallel._utils import _get_global_rank, _get_device_num, _get_parallel_mode
209 …if _get_parallel_mode() == ParallelMode.DATA_PARALLEL and context.get_context("device_target") == …
211 elif _get_parallel_mode() == ParallelMode.DATA_PARALLEL \
214 elif _get_parallel_mode() in (ParallelMode.STAND_ALONE, ParallelMode.HYBRID_PARALLEL):
215 … raise RuntimeError("Parallel optimizer is not supported in {}.".format(_get_parallel_mode()))
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 25 from mindspore.parallel._utils import _get_parallel_mode, _get_full_batch
250 parallel_mode = _get_parallel_mode()
348 parallel_mode = _get_parallel_mode()
508 parallel_mode = _get_parallel_mode()
|
| D | thor_layer.py | 29 from mindspore.parallel._utils import _get_parallel_mode, _get_full_batch
749 parallel_mode = _get_parallel_mode()
879 parallel_mode = _get_parallel_mode()
|
| /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
133 parallel_mode = _get_parallel_mode()
293 parallel_mode = _get_parallel_mode()
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | cell_wrapper.py | 19 _get_parallel_mode, _get_enable_parallel_optimizer)
345 self.parallel_mode = _get_parallel_mode()
|
| /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
308 parallel_mode = _get_parallel_mode()
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 23 from ..parallel._utils import _get_parallel_mode
323 …f context.get_context("mode") == context.PYNATIVE_MODE and "auto_parallel" in _get_parallel_mode():
|