Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc63 bool ParameterAggregator::ReInitForUpdatingHyperParams(size_t aggr_threshold) { in ReInitForUpdatingHyperParams() argument
64 required_push_count_ = aggr_threshold; in ReInitForUpdatingHyperParams()
65 required_pull_count_ = aggr_threshold; in ReInitForUpdatingHyperParams()
67 [aggr_threshold](auto aggregation_kernel) { in ReInitForUpdatingHyperParams()
69 … return !aggregation_kernel.first->ReInitForUpdatingHyperParams(aggr_threshold); in ReInitForUpdatingHyperParams()
Dexecutor.h55 bool ReInitForUpdatingHyperParams(size_t aggr_threshold);
Dparameter_aggregator.h72 bool ReInitForUpdatingHyperParams(size_t aggr_threshold);
Dexecutor.cc54 bool Executor::ReInitForUpdatingHyperParams(size_t aggr_threshold) { in ReInitForUpdatingHyperParams() argument
55 aggregation_count_ = aggr_threshold; in ReInitForUpdatingHyperParams()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Dfed_avg_kernel.h181 bool ReInitForUpdatingHyperParams(size_t aggr_threshold) override { in ReInitForUpdatingHyperParams() argument
182 done_count_ = aggr_threshold; in ReInitForUpdatingHyperParams()