Home
last modified time | relevance | path

Searched refs:base_chunk_elts (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.cc58 int64 base_chunk_elts = (total_elts + (num_chunks - 1)) / num_chunks; in AlignedChunkElts() local
59 if (EIGEN_MAX_ALIGN_BYTES == 0) return base_chunk_elts; in AlignedChunkElts()
63 return base_chunk_elts; in AlignedChunkElts()
72 int64 chunk_bytes = base_chunk_elts * elt_bytes; in AlignedChunkElts()
78 base_chunk_elts += (diff / elt_bytes); in AlignedChunkElts()
79 DCHECK_EQ(0, ((base_chunk_elts * elt_bytes) % EIGEN_MAX_ALIGN_BYTES)) in AlignedChunkElts()
82 << " base_chunk_elts=" << base_chunk_elts << " elt_bytes=" << elt_bytes; in AlignedChunkElts()
83 return base_chunk_elts; in AlignedChunkElts()