Searched refs:num_chunks_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 86 num_chunks_(num_chunks), in CollectiveAdapterImpl()
90 ? AlignedChunkElts(sizeof(T), total_elts_, num_chunks_) in CollectiveAdapterImpl()
91 : total_elts_ / num_chunks_), in CollectiveAdapterImpl()
95 DCHECK_EQ(total_elts_, num_chunks_ * chunk_elts_); in CollectiveAdapterImpl()
122 DCHECK_LT(i, num_chunks_); in ChunkElts()
149 " num_chunks ", num_chunks_, " total_elts ", total_elts_, " chunk_elts", in DebugString()
174 const int64 num_chunks_; member in tensorflow::__anon8df3d5d70111::CollectiveAdapterImpl
|