Searched refs:chunk_elts_ (Results 1 – 1 of 1) sorted by relevance
102 chunk_elts_(align_chunks in CollectiveAdapterImpl()108 DCHECK_EQ(total_elts_, num_chunks_ * chunk_elts_); in CollectiveAdapterImpl()110 DCHECK_GT(chunk_elts_, 0); in CollectiveAdapterImpl()136 const T* chunk_start = std::min(data_end_, data_start_ + i * chunk_elts_); in ChunkElts()137 const T* chunk_end = std::min(data_end_, chunk_start + chunk_elts_); in ChunkElts()145 int64_t start = chunk_elts_ * i; in ChunkAlias()165 chunk_elts_, " value ", in DebugString()188 const int64_t chunk_elts_; member in tensorflow::__anonb82aea130211::CollectiveAdapterImpl