Searched refs:pdist_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance
96 __global__ static void pdist_kernel_cuda_impl(scalar_t * result, const scalar_t * self, const int64… in pdist_kernel_cuda_impl() function250 auto impl_fptr = pdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::p>; in pdist_forward_kernel_impl()252 impl_fptr = pdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::zero>; in pdist_forward_kernel_impl()254 impl_fptr = pdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::one>; in pdist_forward_kernel_impl()256 impl_fptr = pdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::two>; in pdist_forward_kernel_impl()258 impl_fptr = pdist_kernel_cuda_impl<scalar_t, dists<scalar_t>::inf>; in pdist_forward_kernel_impl()