Searched refs:opmath_jitted_gpu_kernel_with_scalars (Results 1 – 16 of 16) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | legendre_polynomial_p.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<legendre_polynomial_p_name, scalar_t, scalar_t>(iterator, le… in legendre_polynomial_p_kernel_cuda()
|
| D | hermite_polynomial_h.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<hermite_polynomial_h_name, scalar_t, scalar_t>(iterator, her… in hermite_polynomial_h_kernel_cuda()
|
| D | chebyshev_polynomial_t.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<chebyshev_polynomial_t_name, scalar_t, scalar_t>(iterator, c… in chebyshev_polynomial_t_kernel_cuda()
|
| D | shifted_chebyshev_polynomial_t.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<shifted_chebyshev_polynomial_t_name, scalar_t, scalar_t>(ite… in shifted_chebyshev_polynomial_t_kernel_cuda()
|
| D | laguerre_polynomial_l.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<laguerre_polynomial_l_name, scalar_t, scalar_t>(iterator, la… in laguerre_polynomial_l_kernel_cuda()
|
| D | shifted_chebyshev_polynomial_v.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<shifted_chebyshev_polynomial_v_name, scalar_t, scalar_t>(ite… in shifted_chebyshev_polynomial_v_kernel_cuda()
|
| D | hermite_polynomial_he.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<hermite_polynomial_he_name, scalar_t, scalar_t>(iterator, he… in hermite_polynomial_he_kernel_cuda()
|
| D | shifted_chebyshev_polynomial_w.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<shifted_chebyshev_polynomial_w_name, scalar_t, scalar_t>(ite… in shifted_chebyshev_polynomial_w_kernel_cuda()
|
| D | chebyshev_polynomial_w.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<chebyshev_polynomial_w_name, scalar_t, scalar_t>(iterator, c… in chebyshev_polynomial_w_kernel_cuda()
|
| D | shifted_chebyshev_polynomial_u.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<shifted_chebyshev_polynomial_u_name, scalar_t, scalar_t>(ite… in shifted_chebyshev_polynomial_u_kernel_cuda()
|
| D | chebyshev_polynomial_v.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<chebyshev_polynomial_v_name, scalar_t, scalar_t>(iterator, c… in chebyshev_polynomial_v_kernel_cuda()
|
| D | chebyshev_polynomial_u.cu | 18 …opmath_jitted_gpu_kernel_with_scalars<chebyshev_polynomial_u_name, scalar_t, scalar_t>(iterator, c… in chebyshev_polynomial_u_kernel_cuda()
|
| D | ZetaKernel.cu | 22 opmath_jitted_gpu_kernel_with_scalars</*name=*/zeta_name, in zeta_kernel_cuda()
|
| D | BinaryMulKernel.cu | 29 opmath_jitted_gpu_kernel_with_scalars<mul_name, scalar_t, scalar_t>( in mul_kernel_cuda()
|
| D | BinaryDivTrueKernel.cu | 27 opmath_jitted_gpu_kernel_with_scalars<div_name, scalar_t, scalar_t>( in div_true_kernel_cuda()
|
| D | JitLoops.cuh | 162 void opmath_jitted_gpu_kernel_with_scalars(TensorIteratorBase& iter, const std::string& f) { in opmath_jitted_gpu_kernel_with_scalars() function
|