Searched refs:num_chunks_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 97 num_chunks_(num_chunks), in CollectiveAdapterImpl()
101 ? AlignedChunkElts(sizeof(T), total_elts_, num_chunks_) in CollectiveAdapterImpl()
102 : total_elts_ / num_chunks_), in CollectiveAdapterImpl()
106 DCHECK_EQ(total_elts_, num_chunks_ * chunk_elts_); in CollectiveAdapterImpl()
133 DCHECK_LT(i, num_chunks_); in ChunkElts()
162 " num_chunks ", num_chunks_, " total_elts ", total_elts_, " chunk_elts", in DebugString()
183 const int64 num_chunks_; member in tensorflow::__anon2cb66ace0211::CollectiveAdapterImpl
|