Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h123 void set_enable_parallel_optimizer(bool enable_parallel_optimizer) { in set_enable_parallel_optimizer()
126 bool enable_parallel_optimizer() const { return enable_parallel_optimizer_; } in enable_parallel_optimizer() function
Dstep_parallel.cc1684 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in SetParallelShape() local
/third_party/mindspore/tests/ut/python/parallel/
Dtest_shared_param_and_mix_precision.py60 …el_compile_net(mode, dev_num, net, strategy1=None, strategy2=None, enable_parallel_optimizer=False, argument
/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py471 def set_enable_parallel_optimizer(self, enable_parallel_optimizer): argument
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc118 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in InsertVirtualAssignAdd() local