Searched refs:smooth_l1_kernel_cuda (Results 1 – 1 of 1) sorted by relevance
15 void smooth_l1_kernel_cuda(TensorIteratorBase& iter, double beta) { in smooth_l1_kernel_cuda() function72 REGISTER_DISPATCH(smooth_l1_stub, &smooth_l1_kernel_cuda);