Searched refs:slice_elems (Results 1 – 1 of 1) sorted by relevance
42 SliceIndex slice_elems, in HandleCopies() argument51 slice_elems = static_slice_elems; in HandleCopies()54 const size_t slice_bytes = slice_elems * sizeof(T); in HandleCopies()93 out_base + (batch_idx * indices_size + indices_idx) * slice_elems, in HandleCopies()96 slice_elems, in HandleCopies()108 batch_size * indices_size, slice_elems * sizeof(T), work); in HandleCopies()