Searched defs:gemv (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 1817 void gemv<c10::complex<double>>(CUDABLAS_GEMV_ARGTYPES(c10::complex<double>)) { in gemv() function
1831 void gemv<c10::complex<float>>(CUDABLAS_GEMV_ARGTYPES(c10::complex<float>)) { in gemv() function
1848 void gemv<double>(CUDABLAS_GEMV_ARGTYPES(double)) { in gemv() function
1860 void gemv<float>(CUDABLAS_GEMV_ARGTYPES(float)) { in gemv() function
1875 void gemv<at::Half>(CUDABLAS_GEMV_ARGTYPES(at::Half)) { in gemv() function
1908 void gemv<at::BFloat16>(CUDABLAS_GEMV_ARGTYPES(at::BFloat16)) { in gemv() function
|
| H A D | CUDABlas.h | 237 inline void gemv(CUDABLAS_GEMV_ARGTYPES(Dtype)) { in gemv() function
|
| /aosp_15_r20/external/eigen/bench/perf_monitoring/ |
| H A D | gemvt.cpp | 4 void gemv(const Mat &A, Vec &B, const Vec &C) in gemv() function
|
| H A D | gemv.cpp | 4 void gemv(const Mat &A, const Vec &B, Vec &C) in gemv() function
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | benchmark.py | 208 def gemv(W, x): function
|
| /aosp_15_r20/external/eigen/blas/ |
| H A D | level2_impl.h | 26 int EIGEN_BLAS_FUNC(gemv)(const char *opa, const int *m, const int *n, const RealScalar *palpha, in EIGEN_BLAS_FUNC() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BlasKernel.cpp | 746 void gemv(char trans, int64_t m, int64_t n, scalar_t alpha, const scalar_t *a, int64_t lda, const s… in gemv() function
|