Searched refs:batch_offsets_from_efficient_size (Results 1 – 1 of 1) sorted by relevance
122 Tensor batch_offsets_from_efficient_size(const Tensor& ef_sizes) { in batch_offsets_from_efficient_size() function168 Tensor offsets = batch_offsets_from_efficient_size(nt_sizes); in NestedTensor_to_padded_tensor_cuda()