Searched refs:grads_accum_counter_ (Results 1 – 2 of 2) sorted by relevance
162 grads_accum_counter_[key] = 0; in InitGrad()261 grads_accum_counter_[key] = 0; in InitEmbeddingTable()347 grads_accum_counter_[key] += 1; in AccumGrad()348 if (grads_accum_counter_[key] == worker_num_) { in AccumGrad()445 return grads_accum_counter_.size() > 0 && grad_accum_count_ == grads_accum_counter_.size(); in ReadyForUpdateWeights()468 for (auto iter = grads_accum_counter_.begin(); iter != grads_accum_counter_.end(); iter++) { in ResetGradAccumCount()469 grads_accum_counter_[iter->first] = 0; in ResetGradAccumCount()
169 std::unordered_map<Key, size_t> grads_accum_counter_; variable