Home
last modified time | relevance | path

Searched defs:enable_parallel_optimizer (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_shared_param_and_mix_precision.py66 …el_compile_net(mode, dev_num, net, strategy1=None, strategy2=None, enable_parallel_optimizer=False, argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/utils/
Dparallel_context.h173 void set_enable_parallel_optimizer(bool enable_parallel_optimizer) { in set_enable_parallel_optimizer()
176 bool enable_parallel_optimizer() const { return enable_parallel_optimizer_; } in enable_parallel_optimizer() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_auto_parallel_context.py821 def set_enable_parallel_optimizer(self, enable_parallel_optimizer): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc406 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in InsertVirtualAssignAdd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc1645 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in SetParallelShape() local