UniqueCub.cuh 352 B

12345678910111213141516
  1. #include <ATen/core/Tensor.h>
  2. namespace at {
  3. namespace native {
  4. namespace internal {
  5. template <typename scalar_t>
  6. std::tuple<Tensor, Tensor, Tensor> unique_cuda_template(
  7. const Tensor& self,
  8. const bool consecutive,
  9. const bool return_inverse,
  10. const bool return_counts);
  11. } // namespace internal
  12. } // namespace at
  13. } // namespace native