Searched refs:done_count_ (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | dense_grad_accum_kernel.h | 83 if (accum_count_ > done_count_) { in Launch()
84 MS_LOG(ERROR) << "accum_count_ should not be greater than done_count_ " << done_count_; in Launch()
87 if (accum_count_ == done_count_) { in Launch()
89 grad_addr[i] /= done_count_; in Launch()
97 bool IsAggregationDone() { return accum_count_ >= done_count_; } in IsAggregationDone()
|
| D | aggregation_kernel.h | 38 AggregationKernel() : name_(""), done_(false), done_count_(0), accum_count_(0) {} in AggregationKernel()
53 void set_done_count(size_t count) { done_count_ = count; } in set_done_count()
90 size_t done_count_; variable
|
| D | fed_avg_kernel.h | 112 …DistributedCountService::GetInstance().RegisterCounter(name_, done_count_, {first_cnt_handler_, la… in InitKernel()
177 …DistributedCountService::GetInstance().RegisterCounter(name_, done_count_, {first_cnt_handler_, la… in ReInitForScaling()
182 done_count_ = aggr_threshold; in ReInitForUpdatingHyperParams()
183 if (!DistributedCountService::GetInstance().ReInitCounter(name_, done_count_)) { in ReInitForUpdatingHyperParams()
|