Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistanceKernel.cu188 __global__ static void cdist_kernel_cuda_impl(scalar_t * result, const scalar_t * x1, const scalar_… in cdist_kernel_cuda_impl() function
224 auto impl_fptr = cdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::p>; in cdist_kernel_impl()
226 impl_fptr = cdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::zero>; in cdist_kernel_impl()
228 impl_fptr = cdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::one>; in cdist_kernel_impl()
230 impl_fptr = cdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::two>; in cdist_kernel_impl()
232 impl_fptr = cdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::inf>; in cdist_kernel_impl()