Searched refs:argmin_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance
21 void argmin_kernel_cuda_impl(TensorIterator& iter) { in argmin_kernel_cuda_impl() function34 argmin_kernel_cuda_impl<at::Half, float>(iter); in argmin_kernel_cuda()36 argmin_kernel_cuda_impl<at::BFloat16, float>(iter); in argmin_kernel_cuda()39 argmin_kernel_cuda_impl<scalar_t>(iter); in argmin_kernel_cuda()