Searched refs:opmath_gpu_kernel_with_scalars (Results 1 – 2 of 2) sorted by relevance
167 void opmath_gpu_kernel_with_scalars(TensorIteratorBase& iter, const func_t& f) { in opmath_gpu_kernel_with_scalars() function247 opmath_gpu_kernel_with_scalars<arg1_t, arg2_t, return_t, func_t>(iter, f); in gpu_kernel_with_scalars()
31 opmath_gpu_kernel_with_scalars<scalar_t>(iter, DivFunctor<opmath_t>()); in div_true_kernel_cuda()