Searched refs:elt_bytes (Results 1 – 2 of 2) sorted by relevance
55 int64 CollectiveAdapter::AlignedChunkElts(int64 elt_bytes, int64 total_elts, in AlignedChunkElts() argument60 if (EIGEN_MAX_ALIGN_BYTES <= elt_bytes) { in AlignedChunkElts()62 DCHECK_EQ(0, elt_bytes % EIGEN_MAX_ALIGN_BYTES); in AlignedChunkElts()67 DCHECK_EQ(0, EIGEN_MAX_ALIGN_BYTES % elt_bytes) in AlignedChunkElts()70 << " elt_bytes=" << elt_bytes; in AlignedChunkElts()72 int64 chunk_bytes = base_chunk_elts * elt_bytes; in AlignedChunkElts()77 DCHECK_EQ(0, diff % 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()
78 static int64 AlignedChunkElts(int64 elt_bytes, int64 total_elts,