Searched refs:ChunkElts (Results 1 – 1 of 1) sorted by relevance
132 inline int64 ChunkElts(int i) const { in ChunkElts() function in tensorflow::__anon2cb66ace0211::CollectiveAdapterImpl139 int64 ChunkBytes(int i) const override { return sizeof(T) * ChunkElts(i); } in ChunkBytes()144 int64 num_elts = ChunkElts(i); in ChunkAlias()156 return Tensor(allocator_, dt_, {ChunkElts(i)}, empty); in TempChunk()