Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_ps_context.py70 "worker_step_num_per_iteration": ps_context().set_worker_step_num_per_iteration,
/third_party/mindspore/mindspore/ccsrc/ps/
Dps_context.h162 void set_worker_step_num_per_iteration(uint64_t worker_step_num_per_iteration);
Dps_context.cc400 void PSContext::set_worker_step_num_per_iteration(uint64_t worker_step_num_per_iteration) { in set_worker_step_num_per_iteration() function in mindspore::ps::PSContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc409 .def("set_worker_step_num_per_iteration", &PSContext::set_worker_step_num_per_iteration, in PYBIND11_MODULE()