Searched refs:chebyshev_polynomial_u_forward (Results 1 – 4 of 4) sorted by relevance
23 return chebyshev_polynomial_u_forward<scalar_t, true>(x, n); in chebyshev_polynomial_u_kernel_cuda()
1968 T chebyshev_polynomial_u_forward(T x, int64_t n) {2011 T chebyshev_polynomial_u_forward(T x, T n) {2012 return chebyshev_polynomial_u_forward(x, static_cast<int64_t>(n));
2873 inline C10_HOST_DEVICE T chebyshev_polynomial_u_forward(T x, int64_t n) { in chebyshev_polynomial_u_forward() function2916 inline C10_HOST_DEVICE T chebyshev_polynomial_u_forward(T x, T n) { in chebyshev_polynomial_u_forward() function2917 return chebyshev_polynomial_u_forward(x, static_cast<int64_t>(n)); in chebyshev_polynomial_u_forward()
1273 return chebyshev_polynomial_u_forward(x, n); in chebyshev_polynomial_u_kernel()