Home
last modified time | relevance | path

Searched refs:set_client_epoch_num (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/parallel/
D_ps_context.py67 "client_epoch_num": ps_context().set_client_epoch_num,
/third_party/mindspore/mindspore/ccsrc/ps/
Dps_context.h152 void set_client_epoch_num(uint64_t client_epoch_num);
Dps_context.cc388 void PSContext::set_client_epoch_num(uint64_t client_epoch_num) { client_epoch_num_ = client_epoch_… in set_client_epoch_num() function in mindspore::ps::PSContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc401 ….def("set_client_epoch_num", &PSContext::set_client_epoch_num, "Set federated learning client epoc… in PYBIND11_MODULE()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Diteration.cc705 ps::PSContext::instance()->set_client_epoch_num(item.value().get<uint64_t>()); in UpdateHyperParams()