Searched refs:pdist_backward_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance
155 __global__ static void pdist_backward_kernel_cuda_impl(scalar_t * buffer, const scalar_t * grad, co… in pdist_backward_kernel_cuda_impl() function288 auto impl_fptr = pdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::p>; in pdist_backward_kernel_impl()290 impl_fptr = pdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::one>; in pdist_backward_kernel_impl()292 impl_fptr = pdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::lt_two>; in pdist_backward_kernel_impl()294 impl_fptr = pdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::two>; in pdist_backward_kernel_impl()296 impl_fptr = pdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::inf>; in pdist_backward_kernel_impl()