Home
last modified time | relevance | path

Searched refs:set_parallel_mode (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py180 def set_parallel_mode(self, parallel_mode): member in _AutoParallelContext
191 ret = self._context_handle.set_parallel_mode(parallel_mode)
638 "parallel_mode": auto_parallel_context().set_parallel_mode,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h89 bool set_parallel_mode(const std::string &parallel_mode);
Dcontext.cc108 bool ParallelContext::set_parallel_mode(const std::string &parallel_mode) { in set_parallel_mode() function in mindspore::parallel::ParallelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc150 .def("set_parallel_mode", &ParallelContext::set_parallel_mode, "Set parallel mode.") in PYBIND11_MODULE()