Searched refs:src_chip_size (Results 1 – 1 of 1) sorted by relevance
239 int64 src_chip_size = 1; in CopyContiguousSlices() local242 src_chip_size *= src.dim_size(i); in CopyContiguousSlices()248 if (src_chip_size != dst_chip_size) { in CopyContiguousSlices()255 if (src_chip_size == 0 && dst_chip_size == 0) { in CopyContiguousSlices()270 const T* src_p = src.base<T>() + (src_chip_size * src_offset); \ in CopyContiguousSlices()272 HandleSliceToElement<T>(src_p, dst_p, src_chip_size * num_slices); \ in CopyContiguousSlices()