xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/UniqueCub.cuh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <ATen/core/Tensor.h>
2 
3 namespace at {
4 namespace native {
5 namespace internal {
6 
7 template <typename scalar_t>
8 std::tuple<Tensor, Tensor, Tensor> unique_cuda_template(
9     const Tensor& self,
10     const bool consecutive,
11     const bool return_inverse,
12     const bool return_counts);
13 
14 } // namespace internal
15 } // namespace at
16 } // namespace native
17