Searched defs:new_offsets (Results 1 – 2 of 2) sorted by relevance
98 auto new_offsets = offsets.clone(); in chunk_nested_tensor() local154 auto new_offsets = offsets.clone(); in split_with_sizes_nested() local
452 auto new_offsets = at::empty({ntensors}, TensorOptions().dtype(kLong)); in select_nested() local