Searched refs:cdist_backward_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance
123 __global__ static void cdist_backward_kernel_cuda_impl(scalar_t * buffer, const scalar_t * grad, co… in cdist_backward_kernel_cuda_impl() function337 auto impl_fptr = cdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::p>; in cdist_backward_kernel_impl()339 impl_fptr = cdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::one>; in cdist_backward_kernel_impl()341 impl_fptr = cdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::lt_two>; in cdist_backward_kernel_impl()343 impl_fptr = cdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::two>; in cdist_backward_kernel_impl()345 impl_fptr = cdist_backward_kernel_cuda_impl<scalar_t, dists<scalar_t>::inf>; in cdist_backward_kernel_impl()