Home
last modified time | relevance | path

Searched refs:max_values_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduceMaxValuesKernel.cu28 void max_values_kernel_cuda_impl(TensorIterator& iter) { in max_values_kernel_cuda_impl() function
38 max_values_kernel_cuda_impl<scalar_t>(iter); in max_values_kernel_cuda()
55 max_values_kernel_cuda_impl<scalar_t>(iter); in max_all_launch_kernel()