Searched refs:elt_bytes (Results 1 – 2 of 2) sorted by relevance
44 int64 CollectiveAdapter::AlignedChunkElts(int64 elt_bytes, int64 total_elts, in AlignedChunkElts() argument49 if (EIGEN_MAX_ALIGN_BYTES <= elt_bytes) { in AlignedChunkElts()51 DCHECK_EQ(0, elt_bytes % EIGEN_MAX_ALIGN_BYTES); in AlignedChunkElts()56 DCHECK_EQ(0, EIGEN_MAX_ALIGN_BYTES % elt_bytes) in AlignedChunkElts()59 << " elt_bytes=" << elt_bytes; in AlignedChunkElts()61 int64 chunk_bytes = base_chunk_elts * elt_bytes; in AlignedChunkElts()66 DCHECK_EQ(0, diff % elt_bytes); in AlignedChunkElts()67 base_chunk_elts += (diff / elt_bytes); in AlignedChunkElts()68 DCHECK_EQ(0, ((base_chunk_elts * elt_bytes) % EIGEN_MAX_ALIGN_BYTES)) in AlignedChunkElts()71 << " base_chunk_elts=" << base_chunk_elts << " elt_bytes=" << elt_bytes; in AlignedChunkElts()
76 static int64 AlignedChunkElts(int64 elt_bytes, int64 total_elts,