Searched refs:gemv_use_fast_path (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BlasKernel.cpp | 140 bool gemv_use_fast_path(C10_UNUSED char trans, C10_UNUSED int64_t m, in gemv_use_fast_path() function
163 template bool gemv_use_fast_path<scalar_t>(char trans, int64_t m, int64_t n, scalar_t alpha, int64_…
190 bool gemv_use_fast_path<float>(C10_UNUSED char trans, int64_t m, int64_t n, C10_UNUSED float alpha,… in gemv_use_fast_path() function
197 bool gemv_use_fast_path<double>(C10_UNUSED char trans, int64_t m, int64_t n, C10_UNUSED double alph… in gemv_use_fast_path() function
198 return gemv_use_fast_path<float>(trans, m, n, (float)alpha, lda, incx, (float)beta, incy); in gemv_use_fast_path()
227 bool gemv_use_fast_path<at::Half>( in gemv_use_fast_path() function
241 bool gemv_use_fast_path<at::BFloat16>( in gemv_use_fast_path() function
750 if (blas_impl::gemv_use_fast_path<scalar_t>(trans, m, n, alpha, lda, incx, beta, incy)) { in gemv()
|