Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py242 def set_run_phase(self, phase): member in _CostModelContext
258 self._context_handle.set_run_phase(phase)
511 "run_phase": cost_model_context().set_run_phase,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h144 void set_run_phase(int64_t);
Dcostmodel_context.cc253 void CostModelContext::set_run_phase(int64_t phase) { in set_run_phase() function in mindspore::parallel::CostModelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc237 .def("set_run_phase", &CostModelContext::set_run_phase, "Set the flag run_phase.") in PYBIND11_MODULE()