Searched refs:num_chunks_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 88 num_chunks_(num_chunks), in CollectiveAdapterImpl()
92 ? AlignedChunkElts(sizeof(T), total_elts_, num_chunks_) in CollectiveAdapterImpl()
93 : total_elts_ / num_chunks_), in CollectiveAdapterImpl()
97 DCHECK_EQ(total_elts_, num_chunks_ * chunk_elts_); in CollectiveAdapterImpl()
124 DCHECK_LT(i, num_chunks_); in ChunkElts()
152 " num_chunks ", num_chunks_, " total_elts ", total_elts_, " chunk_elts", in DebugString()
173 const int64 num_chunks_; member in tensorflow::__anon509782990111::CollectiveAdapterImpl
|