Searched refs:ChunkElts (Results 1 – 1 of 1) sorted by relevance
121 inline int64 ChunkElts(int i) const { in ChunkElts() function in tensorflow::__anon8df3d5d70111::CollectiveAdapterImpl128 int64 ChunkBytes(int i) const override { return sizeof(T) * ChunkElts(i); } in ChunkBytes()133 int64 num_elts = ChunkElts(i); in ChunkAlias()143 return Tensor(allocator_, dt_, {ChunkElts(i)}, empty); in TempChunk()