Searched refs:should_use_small_sort (Results 1 – 3 of 3) sorted by relevance
9 inline bool should_use_small_sort(const TensorBase &self, int64_t dim) { in should_use_small_sort() function
72 if (should_use_small_sort(values, dim)) { in TORCH_IMPL_FUNC()
73 if (should_use_small_sort(self, dim)) { in sort_cuda_kernel()