Home
last modified time | relevance | path

Searched defs:parallel_mode (Results 1 – 25 of 44) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/profiler/
Dparallel_strategy_profiling.cc60 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in IsProfilingParallelStrategyEnabled() local
86 std::string parallel_mode = parallel_context->parallel_mode(); in GetProfilingParallel() local
/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_reshape.py85 def reshape_common(parallel_mode, strategy0, strategy1, strategy2, strategy_loss, search_mode="dyna… argument
424 def reshape_common2(parallel_mode, net): argument
566 def test_flatten_reshape(parallel_mode="auto_parallel", search_mode="dynamic_programming"): argument
590 def test_flatten_reshape2(parallel_mode="auto_parallel", search_mode="dynamic_programming"): argument
637 def test_flatten_reshape3(parallel_mode="auto_parallel", search_mode="dynamic_programming"): argument
671 def test_flatten_reshape4(parallel_mode="semi_auto_parallel"): argument
Dtest_fillv2.py40 def compile_graph(net, device_num, parallel_mode, y, search_mode="recursive_programming"): argument
Dtest_prelu_cell.py70 def reshape_common(parallel_mode): argument
Dtest_gpu_dropout.py65 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): argument
Dtest_gather_batch_dims.py75 def compile_graph(net, device_num, parallel_mode, x, y): argument
Dtest_sparse_gather_v2.py70 def compile_net(net, index_shape, emb_shape, device_num=8, parallel_mode="semi_auto_parallel", argument
Dtest_index_select.py91 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): argument
Dtest_auto_parallel_reshape.py86 def compile_graph(net, parallel_mode, device_num, x, search_mode="dynamic_programming"): argument
93 def compile_graph_two_input(net, parallel_mode, device_num, x, y, search_mode="dynamic_programming"… argument
Dtest_softmax_gather_net.py73 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): argument
Dtest_gather_v2.py87 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): argument
Dtest_reshape_unexpand.py56 def compile_net(net, input_data, dev_num=8, parallel_mode="semi_auto_parallel", search_mode="dynami… argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dgradients_allreduce_depend_last_send.cc29 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in Run() local
Doptimize_gradients_allreduce_overlap.cc413 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_utils.py342 def _device_number_check(parallel_mode, device_number): argument
355 def _parameter_broadcast_check(parallel_mode, parameter_broadcast): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline_split.cc217 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in PipelineSplit() local
316 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in ParallelVirtualDataset() local
Dcompile_cache_manager.cc436 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in GetCachedFuncGraph() local
481 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in InitParallelGroupCkptSaveFile() local
Dpass.cc328 bool parallel_mode() { in parallel_mode() function
334 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in parallel_mode() local
1048 auto parallel_mode = parallel_context->parallel_mode(); in PipelineParallelScheduler() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/dynamic_shape/
Ddynamic_shape.cc217 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in IsParallelDynamicShape() local
226 std::string parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in IsSemiOrAutoParallelMode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dparallel_context.cc171 bool ParallelContext::set_parallel_mode(const std::string &parallel_mode) { in set_parallel_mode()
303 std::string parallel_mode = ParallelContext::GetInstance()->parallel_mode(); in ParallelContextCareGraph() local
Dlineage.proto67 optional string parallel_mode = 5; field
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/profiler/parser/
Dprofiler_info.py44 def set_parallel_info(parallel_mode="", stage_num=1): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Darithmetic_simplify.cc89 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ad/
Dgrad.cc223 auto parallel_mode = parallel_context->parallel_mode(); in GradMultiFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_depend_for_grad_comm.cc280 auto parallel_mode = parallel::ParallelContext::GetInstance()->parallel_mode(); in IsStepIn() local

12