Searched refs:group_size_ (Results 1 – 5 of 5) sorted by relevance
104 group_size_(-1), in RingAlg()326 int recv_from_rank = (rf->rank + (group_size_ - 1)) % group_size_; in InitRingField()327 int send_to_rank = (rf->rank + 1) % group_size_; in InitRingField()339 (rf->rank != ((rf->chunk_idx + (group_size_ - 1)) % group_size_)); in InitRingField()342 (rf->rank == ((rf->chunk_idx + (group_size_ - 1)) % group_size_)); in InitRingField()360 (rf->rank != ((rf->chunk_idx + (group_size_ - 1)) % group_size_)); in AdvanceToSecondPass()362 (rf->rank != ((rf->chunk_idx + (group_size_ - 2)) % group_size_)); in AdvanceToSecondPass()365 (rf->rank == ((rf->chunk_idx + (group_size_ - 2)) % group_size_)); in AdvanceToSecondPass()387 int send_to_rank = (rf->rank + 1) % group_size_; in DispatchSend()402 (rf->rank + (group_size_ - 1)) % group_size_); in DispatchRecv()
57 group_size_ = col_params_->group.group_size; in Run()113 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in ContinueAfterInputCopy()121 Tensor group_size_val = ca_->Scalar(group_size_); in ContinueAfterInputCopy()162 rfv_.resize(group_size_ * num_subdivs_); in RunAsyncParts()164 for (int chunk_idx = 0; chunk_idx < group_size_; ++chunk_idx) { in RunAsyncParts()
65 group_size_ = col_params_->group.group_size; in Run()92 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in Run()126 rfv_.resize(group_size_ * num_subdivs_); in RunAsyncParts()128 for (int chunk_idx = 0; chunk_idx < group_size_; ++chunk_idx) { in RunAsyncParts()
113 int group_size_; variable
484 group_size_ = extent_y_;486 group_size_ = extent_x_ * extent_z_;490 const int group = kOne_ ? ind / group_size_ : ind % group_size_;491 const int offset = kOne_ ? ind % group_size_ : ind / group_size_;503 int group_size_;