| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 222 def _device_number_check(parallel_mode, device_number): argument
235 def _parameter_broadcast_check(parallel_mode, parameter_broadcast): argument
|
| D | _auto_parallel_context.py | 180 def set_parallel_mode(self, parallel_mode): argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_reshape.py | 85 def reshape_common(parallel_mode, strategy0, strategy1, strategy2, strategy_loss): argument
425 def reshape_common2(parallel_mode, net): argument
567 def test_flatten_reshape(parallel_mode="auto_parallel"): argument
586 def test_flatten_reshape2(parallel_mode="auto_parallel"): argument
628 def test_flatten_reshape3(parallel_mode="auto_parallel"): argument
657 def test_flatten_reshape4(parallel_mode="semi_auto_parallel"): argument
|
| D | test_prelu_cell.py | 99 def reshape_common(parallel_mode): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | step_allreduce_fusion.cc | 33 std::string parallel_mode = ParallelContext::GetInstance()->parallel_mode(); in StepAllreduceFusion() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 73 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in PipelineSplit() local
|
| D | pass.cc | 217 bool parallel_mode() { in parallel_mode() function
223 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in parallel_mode() local
|
| D | pipeline.cc | 644 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in SaveCompiledGraph() local
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | lineage.proto | 67 optional string parallel_mode = 5; field
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | context.h | 90 std::string parallel_mode() const { return parallel_mode_; } in parallel_mode() function
|
| D | context.cc | 108 bool ParallelContext::set_parallel_mode(const std::string ¶llel_mode) { in set_parallel_mode()
|
| D | step_auto_parallel.cc | 65 std::string parallel_mode = ParallelContext::GetInstance()->parallel_mode(); in StepAutoParallel() local
|
| D | step_parallel.cc | 2928 std::string parallel_mode = ParallelContext::GetInstance()->parallel_mode(); in ParallelInit() local
3065 std::string parallel_mode = ParallelContext::GetInstance()->parallel_mode(); in StepParallel() local
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 56 parallel_mode = 'parallel_mode' variable in LineageMetadata
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 2472 auto parallel_mode = parallel_context->parallel_mode(); in InitAllBucket() local
2514 auto parallel_mode = parallel_context->parallel_mode(); in AddGradAddrToBucket() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 761 const auto ¶llel_mode = parallel_context->parallel_mode(); in CheckPyNativeContext() local
|