Searched refs:set_client_batch_size (Results 1 – 5 of 5) sorted by relevance
68 "client_batch_size": ps_context().set_client_batch_size,
156 void set_client_batch_size(uint64_t client_batch_size);
392 void PSContext::set_client_batch_size(uint64_t client_batch_size) { client_batch_size_ = client_bat… in set_client_batch_size() function in mindspore::ps::PSContext
403 ….def("set_client_batch_size", &PSContext::set_client_batch_size, "Set federated learning client ba… in PYBIND11_MODULE()
709 ps::PSContext::instance()->set_client_batch_size(item.value().get<uint64_t>()); in UpdateHyperParams()