Searched refs:blas_impl (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BlasKernel.cpp | 85 namespace blas_impl { namespace
729 if (blas_impl::scal_use_fast_path<scalar_t>(n, incx)) { in scal()
732 blas_impl::scal_fast_path<scalar_t>(&i_n, &a, x, &i_incx); in scal()
750 if (blas_impl::gemv_use_fast_path<scalar_t>(trans, m, n, alpha, lda, incx, beta, incy)) { in gemv()
757 …blas_impl::gemv_fast_path<scalar_t>(&trans, &i_m, &i_n, &alpha, a, &i_lda, x, &i_incx, &beta, y, &… in gemv()
822 namespace blas_impl { namespace
887 return blas_impl::dot_fast_path(n, x, incx, y, incy); in dot_impl_floating()
889 return blas_impl::dot_naive(n, x, incx, y, incy, std::multiplies<scalar_t>{}); in dot_impl_floating()
892 { return blas_impl::dot_naive(n, x, incx, y, incy, std::multiplies<scalar_t>{}); } in dot_impl_floating()
902 return blas_impl::dot_naive(n, x, incx, y, incy, std::multiplies<scalar_t>{}); in dot_impl()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | BlasKernel.cpp | 12 namespace at::native::blas_impl { namespace
345 …at::native::blas_impl::fp16_gemv_notrans(m, k, 1.0, reinterpret_cast<const float16_t*>(a), lda, re… in gemm_notrans_()
371 return at::native::blas_impl::fp16_dot_with_fp32_arith( in compute_dot()
378 return at::native::blas_impl::bf16_dot_with_fp32_arith(a, b, len); in compute_dot()
392 …at::native::blas_impl::fp16_gemv_trans(k, m, 1.0, reinterpret_cast<const float16_t*>(a), lda, rein… in gemm_transa_()
|