Searched refs:num_subdivs_ (Results 1 – 4 of 4) sorted by relevance
66 num_subdivs_ = static_cast<int>( in Run()68 DCHECK_GT(num_subdivs_, 0); in Run()92 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in Run()126 rfv_.resize(group_size_ * num_subdivs_); in RunAsyncParts()129 for (int subdiv_idx = 0; subdiv_idx < num_subdivs_; ++subdiv_idx) { in RunAsyncParts()130 int rf_index = (chunk_idx * num_subdivs_) + subdiv_idx; in RunAsyncParts()
58 num_subdivs_ = static_cast<int>( in Run()60 CHECK_GT(num_subdivs_, 0); in Run()113 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in ContinueAfterInputCopy()162 rfv_.resize(group_size_ * num_subdivs_); in RunAsyncParts()165 for (int subdiv_idx = 0; subdiv_idx < num_subdivs_; ++subdiv_idx) { in RunAsyncParts()166 int rf_index = (chunk_idx * num_subdivs_) + subdiv_idx; in RunAsyncParts()
114 int num_subdivs_; variable
105 num_subdivs_(-1) {} in RingAlg()318 DCHECK_EQ(field_idx, (chunk_idx * num_subdivs_) + subdiv_idx); in InitRingField()