Searched refs:cdist_backward_kernel_impl (Results 1 – 2 of 2) sorted by relevance
305 void cdist_backward_kernel_impl(Tensor& result, const Tensor& grad, const Tensor& x1, const Tensor&… in cdist_backward_kernel_impl() function363 REGISTER_DISPATCH(cdist_backward_stub, &cdist_backward_kernel_impl);
437 static void cdist_backward_kernel_impl(Tensor& result, const Tensor& grad, const Tensor& x1, const … in cdist_backward_kernel_impl() function449 REGISTER_DISPATCH(cdist_backward_stub, &cdist_backward_kernel_impl);