Searched refs:ChunkElts (Results 1 – 1 of 1) sorted by relevance
134 inline int64_t ChunkElts(int i) const { in ChunkElts() function in tensorflow::__anonb82aea130211::CollectiveAdapterImpl141 int64_t ChunkBytes(int i) const override { return sizeof(T) * ChunkElts(i); } in ChunkBytes()146 int64_t num_elts = ChunkElts(i); in ChunkAlias()158 return Tensor(allocator_, dt_, {ChunkElts(i)}, empty); in TempChunk()