1 #pragma once 2 #include <cstdint> 3 4 namespace at { 5 class TensorBase; 6 } 7 8 namespace at { 9 namespace native { 10 11 void launch_kthvalue_kernel( 12 const TensorBase &values, const TensorBase &indices, 13 const TensorBase &self, int64_t dim, int64_t k); 14 void launch_median_kernel( 15 const TensorBase &vals, const TensorBase &inds, 16 const TensorBase &in, int64_t dim, bool ignore_nan); 17 18 }} // namespace at::native 19