Home
last modified time | relevance | path

Searched refs:threshold_count (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/fl/server/
Dround.h37 … bool check_count = false, size_t threshold_count = 8, bool server_num_as_threshold = false);
60 size_t threshold_count() const;
Dround.cc28 …td::string &name, bool check_timeout, size_t time_window, bool check_count, size_t threshold_count, in Round() argument
34 threshold_count_(threshold_count), in Round()
181 size_t Round::threshold_count() const { return threshold_count_; } in threshold_count() function in mindspore::fl::server::Round
Dparameter_aggregator.cc28 bool ParameterAggregator::Init(const CNodePtr &cnode, size_t threshold_count) { in Init() argument
33 required_push_count_ = threshold_count; in Init()
36 required_pull_count_ = threshold_count; in Init()
Dparameter_aggregator.h66 bool Init(const CNodePtr &cnode, size_t threshold_count = 0);
Dserver.cc205 config.threshold_count, config.server_num_as_threshold); in InitIteration()
208 << ", threshold: " << config.threshold_count in InitIteration()
402 round_kernel->InitKernel(round->threshold_count()); in RegisterRoundKernel()
Dcommon.h57 size_t threshold_count = 0; member
Diteration.cc201 …if (!round->ReInitForUpdatingHyperParams(updated_round.threshold_count, updated_round.time_window)… in ReInitForUpdatingHyperParams()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dpush_metrics_kernel.h38 void InitKernel(size_t threshold_count);
Dpush_weight_kernel.h38 void InitKernel(size_t threshold_count) override;
Dupdate_model_kernel.h44 void InitKernel(size_t threshold_count) override;
Dstart_fl_job_kernel.h38 void InitKernel(size_t threshold_count) override;
Dround_kernel.h59 virtual void InitKernel(size_t threshold_count) = 0;
Dupdate_model_kernel.cc27 void UpdateModelKernel::InitKernel(size_t threshold_count) { in InitKernel() argument
41 LocalMetaStore::GetInstance().put_value(kCtxUpdateModelThld, threshold_count); in InitKernel()