Searched refs:ParameterAggregator (Results 1 – 4 of 4) sorted by relevance
28 bool ParameterAggregator::Init(const CNodePtr &cnode, size_t threshold_count) { in Init()50 bool ParameterAggregator::ReInitForScaling() { in ReInitForScaling()63 bool ParameterAggregator::ReInitForUpdatingHyperParams(size_t aggr_threshold) { in ReInitForUpdatingHyperParams()78 bool ParameterAggregator::UpdateData(const std::map<std::string, Address> &new_data) { in UpdateData()100 bool ParameterAggregator::LaunchAggregators() { in LaunchAggregators()114 bool ParameterAggregator::LaunchOptimizers() { in LaunchOptimizers()130 AddressPtr ParameterAggregator::Pull() { in Pull()147 AddressPtr ParameterAggregator::GetWeight() { in GetWeight()157 void ParameterAggregator::ResetAggregationStatus() { in ResetAggregationStatus()169 void ParameterAggregator::ResetOptimizingStatus() { optimizing_done_ = false; } in ResetOptimizingStatus()[all …]
50 class ParameterAggregator {52 ParameterAggregator() in ParameterAggregator() function62 ~ParameterAggregator() = default;
124 std::map<std::string, std::shared_ptr<ParameterAggregator>> param_aggrs_;
337 std::shared_ptr<ParameterAggregator> param_aggr = std::make_shared<ParameterAggregator>(); in InitParamAggregator()