| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/ |
| H A D | test_cblas.py | 28 def verify_matmul_add(m, l, n, lib, transa=False, transb=False, dtype="float32"): argument
38 def get_numpy(a, b, bb, transa, transb): argument
93 def verify_quantized_matmul_add(m, l, n, transa=False, transb=False): argument
108 def get_numpy(a, b, bb, transa, transb): argument
150 batch, m, l, n, lib, transa=False, transb=False, iterative=False, dtype="float32" argument
160 def get_numpy(a, b, transa, transb): argument
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/gemm/ |
| H A D | gemm_pack.cpp | 42 dnnl_status_t sgemm_pack_get_size(const char *identifier, const char *transa, in sgemm_pack_get_size()
53 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_bf16bf16f32_pack_get_size()
64 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_s8u8s32_pack_get_size()
75 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_s8s8s32_pack_get_size()
85 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack()
95 dnnl_status_t gemm_bf16bf16f32_pack(const char *identifier, const char *transa, in gemm_bf16bf16f32_pack()
106 dnnl_status_t gemm_s8u8s32_pack(const char *identifier, const char *transa, in gemm_s8u8s32_pack()
116 dnnl_status_t gemm_s8s8s32_pack(const char *identifier, const char *transa, in gemm_s8s8s32_pack()
126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute()
148 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, in gemm_s8u8s32_compute()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/gemm/ |
| H A D | gemm_pack.cpp | 42 dnnl_status_t sgemm_pack_get_size(const char *identifier, const char *transa, in sgemm_pack_get_size()
53 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_bf16bf16f32_pack_get_size()
64 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_s8u8s32_pack_get_size()
75 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in gemm_s8s8s32_pack_get_size()
85 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack()
95 dnnl_status_t gemm_bf16bf16f32_pack(const char *identifier, const char *transa, in gemm_bf16bf16f32_pack()
106 dnnl_status_t gemm_s8u8s32_pack(const char *identifier, const char *transa, in gemm_s8u8s32_pack()
116 dnnl_status_t gemm_s8s8s32_pack(const char *identifier, const char *transa, in gemm_s8s8s32_pack()
126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute()
148 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, in gemm_s8u8s32_compute()
[all …]
|
| /dports/science/xtb/xtb-6.4.1/src/mctc/blas/ |
| H A D | wrap3.f90 | 52 subroutine mctc_sgemm323(amat, bmat, cmat, transa, transb, alpha, beta) argument
56 character(len=1), intent(in), optional :: transa local
77 subroutine mctc_sgemm233(amat, bmat, cmat, transa, transb, alpha, beta) argument
81 character(len=1), intent(in), optional :: transa local
101 subroutine mctc_sgemm332(amat, bmat, cmat, transa, transb, alpha, beta) argument
105 character(len=1), intent(in), optional :: transa local
135 subroutine mctc_dgemm323(amat, bmat, cmat, transa, transb, alpha, beta) argument
139 character(len=1), intent(in), optional :: transa local
160 subroutine mctc_dgemm233(amat, bmat, cmat, transa, transb, alpha, beta) argument
164 character(len=1), intent(in), optional :: transa local
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/contrib/ |
| H A D | test_cblas.py | 22 def verify_matmul_add(m, l, n, transa=False, transb=False, dtype=tvm.float32): argument
32 def get_numpy(a, b, bb, transa, transb): argument
67 def verify_batch_matmul(batch, m, l, n, transa=False, transb=False, iterative=False, dtype=tvm.floa… argument
76 def get_numpy(a, b, transa, transb): argument
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/contrib/ |
| H A D | test_cblas.py | 22 def verify_matmul_add(m, l, n, transa=False, transb=False, dtype=tvm.float32): argument
32 def get_numpy(a, b, bb, transa, transb): argument
67 def verify_batch_matmul(batch, m, l, n, transa=False, transb=False, iterative=False, dtype=tvm.floa… argument
76 def get_numpy(a, b, transa, transb): argument
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/ |
| H A D | mkl.py | 22 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument
55 def matmul_u8s8s32(lhs, rhs, transa=False, transb=False, **kwargs): argument
88 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument
|
| H A D | rocblas.py | 22 def matmul(lhs, rhs, transa=False, transb=False): argument
53 def batch_matmul(lhs, rhs, transa=False, transb=False): argument
|
| H A D | cublas.py | 22 def matmul(lhs, rhs, transa=False, transb=False, dtype=None): argument
55 def batch_matmul(lhs, rhs, transa=False, transb=False, dtype=None): argument
|
| H A D | cblas.py | 22 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument
55 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument
|
| /dports/science/py-chainer/chainer-7.8.0/chainer/functions/math/ |
| H A D | matmul.py | 41 def _matmul(a, b, transa=False, transb=False, transout=False): argument
75 def __init__(self, transa=False, transb=False, transc=False, dtype=None): argument
187 def matmul(a, b, transa=False, transb=False): argument
230 def _batch_matmul(a, b, transa, transb, transout): argument
238 def __init__(self, transa=False, transb=False): argument
275 def __init__(self, transa=False, transb=False): argument
307 def batch_matmul(a, b, transa=False, transb=False): argument
|
| /dports/science/qmcpack/qmcpack-3.11.0/src/Numerics/ |
| H A D | MatrixOperators.h | 47 const char transa = 'N'; in product() local
58 const char transa = 't'; in product_ABt() local
69 const char transa = 'n'; in product_AtB() local
178 const char transa = 'T'; in product() local
189 const char transa = 'N'; in product_Atx() local
200 const char transa = 'T'; in product() local
211 const char transa = 'N'; in product_Atx() local
224 const char transa = 'N'; in product() local
234 const char transa = 'N'; in product() local
249 const char transa = 'N'; in product() local
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/ |
| H A D | gemm_pack.cpp | 106 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in check_pack_get_size_input()
128 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in check_pack_input()
163 dnnl_status_t sgemm_pack_get_size(const char *identifier, const char *transa, in sgemm_pack_get_size()
303 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack()
348 dnnl_status_t gemm_x8x8s32_pack(const char *identifier, const char *transa, in gemm_x8x8s32_pack()
399 dnnl_status_t gemm_s8u8s32_pack(const char *identifier, const char *transa, in gemm_s8u8s32_pack()
407 dnnl_status_t gemm_s8s8s32_pack(const char *identifier, const char *transa, in gemm_s8s8s32_pack()
415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute()
451 dnnl_status_t gemm_x8x8s32_compute(const char *transa, const char *transb, in gemm_x8x8s32_compute()
509 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, in gemm_s8u8s32_compute()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
| H A D | gemm_pack.cpp | 106 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in check_pack_get_size_input()
128 const char *transa, const char *transb, const dim_t *M, const dim_t *N, in check_pack_input()
163 dnnl_status_t sgemm_pack_get_size(const char *identifier, const char *transa, in sgemm_pack_get_size()
303 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack()
348 dnnl_status_t gemm_x8x8s32_pack(const char *identifier, const char *transa, in gemm_x8x8s32_pack()
399 dnnl_status_t gemm_s8u8s32_pack(const char *identifier, const char *transa, in gemm_s8u8s32_pack()
407 dnnl_status_t gemm_s8s8s32_pack(const char *identifier, const char *transa, in gemm_s8s8s32_pack()
415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute()
451 dnnl_status_t gemm_x8x8s32_compute(const char *transa, const char *transb, in gemm_x8x8s32_compute()
509 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, in gemm_s8u8s32_compute()
[all …]
|
| /dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/detail/CPU/ |
| H A D | sparse_cpu.hpp | 32 void csrmv(const char transa, in csrmv()
92 void csrmv(const char transa, in csrmv()
152 void csrmm(const char transa, in csrmm()
238 void csrmm(const char transa, in csrmm()
325 inline static void csrmv(const char transa, in csrmv()
345 inline static void csrmv(const char transa, in csrmv()
365 inline static void csrmv(const char transa, in csrmv()
385 inline static void csrmv(const char transa, in csrmv()
405 inline static void csrmm(const char transa, in csrmm()
428 inline static void csrmm(const char transa, in csrmm()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/contrib/ |
| H A D | cublas.py | 23 def matmul(lhs, rhs, transa=False, transb=False): argument
50 def batch_matmul(lhs, rhs, transa=False, transb=False): argument
|
| H A D | cblas.py | 23 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument
57 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/contrib/ |
| H A D | cublas.py | 23 def matmul(lhs, rhs, transa=False, transb=False): argument
50 def batch_matmul(lhs, rhs, transa=False, transb=False): argument
|
| H A D | cblas.py | 23 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument
57 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument
|
| /dports/science/qmcpack/qmcpack-3.11.0/src/Platforms/CUDA/ |
| H A D | cuBLAS.hpp | 80 const cublasOperation_t& transa, in gemm()
98 const cublasOperation_t& transa, in gemm()
117 const cublasOperation_t& transa, in gemm()
135 const cublasOperation_t& transa, in gemm()
154 const cublasOperation_t& transa, in gemm_batched()
173 const cublasOperation_t& transa, in gemm_batched()
193 const cublasOperation_t& transa, in gemm_batched()
212 const cublasOperation_t& transa, in gemm_batched()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | gemm.cpp | 45 dnnl_status_t dnnl_sgemm(char transa, char transb, dim_t M, dim_t N, dim_t K, in dnnl_sgemm()
56 dnnl_status_t dnnl_gemm_u8s8s32(char transa, char transb, char offsetc, dim_t M, in dnnl_gemm_u8s8s32()
68 dnnl_status_t dnnl_gemm_s8s8s32(char transa, char transb, char offsetc, dim_t M, in dnnl_gemm_s8s8s32()
80 extern "C" dnnl_status_t DNNL_API dnnl_gemm_bf16bf16f32(char transa, in dnnl_gemm_bf16bf16f32()
93 dnnl_status_t dnnl_threadpool_interop_sgemm(char transa, char transb, dim_t M, in dnnl_threadpool_interop_sgemm()
105 dnnl_status_t dnnl_threadpool_interop_gemm_u8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_u8s8s32()
117 dnnl_status_t dnnl_threadpool_interop_gemm_s8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_s8s8s32()
131 char transa, char transb, dim_t M, dim_t N, dim_t K, float alpha, in dnnl_threadpool_interop_gemm_bf16bf16f32()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/contrib/ |
| H A D | rocblas.h | 45 inline Tensor rocblas_matmul(const Tensor& lhs, const Tensor& rhs, bool transa, bool transb) { in rocblas_matmul()
67 inline Tensor rocblas_batch_matmul(const Tensor& lhs, const Tensor& rhs, bool transa, bool transb) { in rocblas_batch_matmul()
|
| H A D | cublas.h | 46 inline Tensor cublas_matmul(const Tensor& lhs, const Tensor& rhs, bool transa, bool transb) { in cublas_matmul()
70 inline Tensor cublas_batch_matmul(const Tensor& lhs, const Tensor& rhs, bool transa, bool transb) { in cublas_batch_matmul()
|
| /dports/math/libflame/libflame-5.2.0/src/base/flamec/blis/2/ |
| H A D | bl1_gemv.c | 13 void bl1_sgemv( trans1_t transa, conj1_t conjx, int m, int n, float* alpha, float* a, int a_rs, int… in bl1_sgemv()
69 void bl1_dgemv( trans1_t transa, conj1_t conjx, int m, int n, double* alpha, double* a, int a_rs, i… in bl1_dgemv()
125 void bl1_cgemv( trans1_t transa, conj1_t conjx, int m, int n, scomplex* alpha, scomplex* a, int a_r… in bl1_cgemv()
255 void bl1_zgemv( trans1_t transa, conj1_t conjx, int m, int n, dcomplex* alpha, dcomplex* a, int a_r… in bl1_zgemv()
387 void bl1_sgemv_blas( trans1_t transa, int m, int n, float* alpha, float* a, int lda, float* x, int … in bl1_sgemv_blas()
420 void bl1_dgemv_blas( trans1_t transa, int m, int n, double* alpha, double* a, int lda, double* x, i… in bl1_dgemv_blas()
453 void bl1_cgemv_blas( trans1_t transa, int m, int n, scomplex* alpha, scomplex* a, int lda, scomplex… in bl1_cgemv_blas()
486 void bl1_zgemv_blas( trans1_t transa, int m, int n, dcomplex* alpha, dcomplex* a, int lda, dcomplex… in bl1_zgemv_blas()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | gemm.cpp | 53 dnnl_status_t dnnl_sgemm(char transa, char transb, dim_t M, dim_t N, dim_t K, in dnnl_sgemm()
65 dnnl_status_t dnnl_gemm_u8s8s32(char transa, char transb, char offsetc, dim_t M, in dnnl_gemm_u8s8s32()
78 dnnl_status_t dnnl_gemm_s8s8s32(char transa, char transb, char offsetc, dim_t M, in dnnl_gemm_s8s8s32()
91 extern "C" dnnl_status_t DNNL_API dnnl_gemm_bf16bf16f32(char transa, in dnnl_gemm_bf16bf16f32()
105 dnnl_status_t dnnl_threadpool_interop_sgemm(char transa, char transb, dim_t M, in dnnl_threadpool_interop_sgemm()
118 dnnl_status_t dnnl_threadpool_interop_gemm_u8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_u8s8s32()
132 dnnl_status_t dnnl_threadpool_interop_gemm_s8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_s8s8s32()
147 char transa, char transb, dim_t M, dim_t N, dim_t K, float alpha, in dnnl_threadpool_interop_gemm_bf16bf16f32()
|