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.cc49 int64 base_chunk_elts = (total_elts + (num_chunks - 1)) / num_chunks; in AlignedChunkElts() local
50 if (EIGEN_MAX_ALIGN_BYTES == 0) return base_chunk_elts; in AlignedChunkElts()
54 return base_chunk_elts; in AlignedChunkElts()
63 int64 chunk_bytes = base_chunk_elts * elt_bytes; in AlignedChunkElts()
69 base_chunk_elts += (diff / elt_bytes); in AlignedChunkElts()
70 DCHECK_EQ(0, ((base_chunk_elts * elt_bytes) % EIGEN_MAX_ALIGN_BYTES)) in AlignedChunkElts()
73 << " base_chunk_elts=" << base_chunk_elts << " elt_bytes=" << elt_bytes; in AlignedChunkElts()
74 return base_chunk_elts; in AlignedChunkElts()