Home
last modified time | relevance | path

Searched refs:transb (Results 1 – 25 of 1227) sorted by relevance

12345678910>>...50

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/
H A Dtest_rocblas.py53 bshape = (batch, n, k) if transb else (batch, k, n)
56 C = lib.batch_matmul(A, B, transa, transb)
59 def get_numpy(a, b, transa, transb): argument
62 if not transb:
89 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=False, transb=True)
90 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=True, transb=False)
91 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=True, transb=True)
95 verify_batch_matmul(128, 512, 512, 64, rocblas, transa=True, transb=True)
99 verify_batch_matmul(128, 512, 64, 512, rocblas, transa=True, transb=True)
103 verify_batch_matmul(128, 64, 128, 128, rocblas, transa=True, transb=True)
[all …]
/dports/math/libflame/libflame-5.2.0/src/blas/3/gemm/front/flamec/
H A DFLA_Gemm_internal.c29 transb, in FLA_Gemm_internal()
57 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal()
59 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal()
63 else if ( transb == FLA_CONJ_TRANSPOSE ) in FLA_Gemm_internal()
68 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal()
70 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal()
74 else if ( transb == FLA_CONJ_TRANSPOSE ) in FLA_Gemm_internal()
79 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal()
81 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal()
90 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal()
[all …]
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/math/
H A Dmatmul.py43 transa, transb = not transb, not transa
47 if transb and b.ndim != 1:
77 self.transb = transb
168 if self.transb:
187 def matmul(a, b, transa=False, transb=False): argument
220 return MatMul(transa=transa, transb=transb).apply((a, b))[0]
230 def _batch_matmul(a, b, transa, transb, transout): argument
233 return _matmul(a, b, transa, transb, transout)
240 self.transb = transb
277 self.transb = transb
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/gemm/
H A Dgemm_pack.cpp47 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in sgemm_pack_get_size()
58 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_bf16bf16f32_pack_get_size()
69 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8u8s32_pack_get_size()
80 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8s8s32_pack_get_size()
90 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in sgemm_pack()
101 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_bf16bf16f32_pack()
111 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8u8s32_pack()
121 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8s8s32_pack()
126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument
132 transa, transb, M, N, K, A, lda, B, ldb, beta, C, ldc); in sgemm_compute()
[all …]
H A Dgemm.cpp49 dnnl_status_t check_gemm_input(const char *transa, const char *transb, in check_gemm_input() argument
66 bool is_packed_b = utils::one_of(*transb, 'P', 'p'); in check_gemm_input()
68 bool is_trans_b = utils::one_of(*transb, 'T', 't'); in check_gemm_input()
92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument
103 && utils::one_of(*transb, 'n', 'N', 't', 'T')) { in extended_sgemm()
105 bool trB = *transb == 't' || *transb == 'T'; in extended_sgemm()
157 bool BisN = (*transb == 'N' || *transb == 'n'); in try_cblas_gemm_s8u8s32()
174 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument
200 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument
229 dnnl_status_t gemm_bf16bf16f32(const char *transa, const char *transb, in gemm_bf16bf16f32() argument
[all …]
H A Dgemm_pack.hpp33 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
38 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
43 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
48 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
57 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
62 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
67 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
71 dnnl_status_t DNNL_API sgemm_compute(const char *transa, const char *transb,
77 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/cpu/gemm/
H A Dgemm_pack.cpp47 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in sgemm_pack_get_size()
58 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_bf16bf16f32_pack_get_size()
69 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8u8s32_pack_get_size()
80 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8s8s32_pack_get_size()
90 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in sgemm_pack()
101 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_bf16bf16f32_pack()
111 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8u8s32_pack()
121 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8s8s32_pack()
126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument
132 transa, transb, M, N, K, A, lda, B, ldb, beta, C, ldc); in sgemm_compute()
[all …]
H A Dgemm.cpp49 dnnl_status_t check_gemm_input(const char *transa, const char *transb, in check_gemm_input() argument
66 bool is_packed_b = utils::one_of(*transb, 'P', 'p'); in check_gemm_input()
68 bool is_trans_b = utils::one_of(*transb, 'T', 't'); in check_gemm_input()
92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument
103 && utils::one_of(*transb, 'n', 'N', 't', 'T')) { in extended_sgemm()
105 bool trB = *transb == 't' || *transb == 'T'; in extended_sgemm()
157 bool BisN = (*transb == 'N' || *transb == 'n'); in try_cblas_gemm_s8u8s32()
174 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument
200 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument
229 dnnl_status_t gemm_bf16bf16f32(const char *transa, const char *transb, in gemm_bf16bf16f32() argument
[all …]
H A Dgemm_pack.hpp33 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
38 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
43 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
48 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
57 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
62 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
67 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
71 dnnl_status_t DNNL_API sgemm_compute(const char *transa, const char *transb,
77 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/
H A Dgemm_pack.cpp113 bool is_transb = utils::one_of(*transb, 'T', 't'); in check_pack_get_size_input()
116 && utils::one_of(*transb, 'T', 't', 'N', 'n') in check_pack_get_size_input()
134 identifier, transa, transb, M, N, K, lda, ldb); in check_pack_input()
174 identifier, transa, transb, M, N, K, lda, ldb); in sgemm_pack_get_size()
192 && utils::one_of(do_a ? *transa : *transb, 'n', 'N') in sgemm_pack_get_size()
212 identifier, transa, transb, M, N, K, lda, ldb); in gemm_bf16bf16f32_pack_get_size()
239 identifier, transa, transb, M, N, K, lda, ldb); in gemm_x8x8s32_pack_get_size()
317 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in sgemm_pack()
365 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in gemm_x8x8s32_pack()
415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument
[all …]
H A Dgemm_pack.hpp34 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
38 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
43 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
48 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
57 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
62 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
66 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
69 dnnl_status_t sgemm_compute(const char *transa, const char *transb,
79 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb,
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/
H A Dgemm_pack.cpp113 bool is_transb = utils::one_of(*transb, 'T', 't'); in check_pack_get_size_input()
116 && utils::one_of(*transb, 'T', 't', 'N', 'n') in check_pack_get_size_input()
134 identifier, transa, transb, M, N, K, lda, ldb); in check_pack_input()
174 identifier, transa, transb, M, N, K, lda, ldb); in sgemm_pack_get_size()
192 && utils::one_of(do_a ? *transa : *transb, 'n', 'N') in sgemm_pack_get_size()
212 identifier, transa, transb, M, N, K, lda, ldb); in gemm_bf16bf16f32_pack_get_size()
239 identifier, transa, transb, M, N, K, lda, ldb); in gemm_x8x8s32_pack_get_size()
317 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in sgemm_pack()
365 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in gemm_x8x8s32_pack()
415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument
[all …]
H A Dgemm_pack.hpp34 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
38 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
43 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
48 const char *transa, const char *transb, const dim_t *M, const dim_t *N,
53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
57 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
62 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
66 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K,
69 dnnl_status_t sgemm_compute(const char *transa, const char *transb,
79 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb,
[all …]
/dports/multimedia/opentoonz/opentoonz-1.5.0/thirdparty/openblas/xianyi-OpenBLAS-e6e87a2/interface/
H A Dgemm.c112 int transa, transb, nrowa, nrowb; local
167 transb = -1;
179 if (transB == 'N') transb = 0;
180 if (transB == 'T') transb = 1;
182 if (transB == 'R') transb = 0;
183 if (transB == 'C') transb = 1;
185 if (transB == 'R') transb = 2;
186 if (transB == 'C') transb = 3;
192 if (transb & 1) nrowb = args.n;
229 int transa, transb;
[all …]
/dports/cad/meshlab/meshlab-Meshlab-2020.05/src/plugins_unsupported/external/GotoBLAS2/interface/
H A Dgemm.c95 int transa, transb, nrowa, nrowb; local
150 transb = -1;
162 if (transB == 'N') transb = 0;
163 if (transB == 'T') transb = 1;
165 if (transB == 'R') transb = 0;
166 if (transB == 'C') transb = 1;
168 if (transB == 'R') transb = 2;
169 if (transB == 'C') transb = 3;
175 if (transb & 1) nrowb = args.n;
212 int transa, transb;
[all …]
/dports/math/gotoblas/GotoBLAS2/interface/
H A Dgemm.c112 int transa, transb, nrowa, nrowb; local
167 transb = -1;
179 if (transB == 'N') transb = 0;
180 if (transB == 'T') transb = 1;
182 if (transB == 'R') transb = 0;
183 if (transB == 'C') transb = 1;
185 if (transB == 'R') transb = 2;
186 if (transB == 'C') transb = 3;
192 if (transb & 1) nrowb = args.n;
229 int transa, transb;
[all …]
/dports/math/openblas/OpenBLAS-0.3.18/interface/
H A Dgemm.c169 int transa, transb, nrowa, nrowb; local
227 transb = -1;
239 if (transB == 'N') transb = 0;
240 if (transB == 'T') transb = 1;
242 if (transB == 'R') transb = 0;
243 if (transB == 'C') transb = 1;
245 if (transB == 'R') transb = 2;
246 if (transB == 'C') transb = 3;
252 if (transb & 1) nrowb = args.n;
294 int transa, transb;
[all …]
/dports/math/libflame/libflame-5.2.0/src/base/flamec/blis/3/
H A Dbl1_gemm.c57 bl1_screate_contigmt( transb, in bl1_sgemm()
105 bl1_toggle_trans( transb ); in bl1_sgemm()
175 bl1_toggle_trans( transb ); in bl1_sgemm()
217 transb, in bl1_sgemm()
249 transb, in bl1_sgemm()
478 transb, in bl1_dgemm()
510 transb, in bl1_dgemm()
810 transb, in bl1_cgemm()
842 transb, in bl1_cgemm()
1148 transb, in bl1_zgemm()
[all …]
/dports/math/libflame/libflame-5.2.0/src/lapack/sol/sylv/front/flamec/
H A DFLA_Sylv_internal.c16 FLA_Error FLA_Sylv_internal( FLA_Trans transa, FLA_Trans transb, FLA_Obj isgn, FLA_Obj A, FLA_Obj B… in FLA_Sylv_internal() argument
21 FLA_Sylv_internal_check( transa, transb, isgn, A, B, C, scale, cntl ); in FLA_Sylv_internal()
29 transb, in FLA_Sylv_internal()
42 ENQUEUE_FLASH_Sylv( transa, transb, isgn, A, B, C, scale, cntl ); in FLA_Sylv_internal()
57 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Sylv_internal()
59 else if ( transb == FLA_TRANSPOSE || transb == FLA_CONJ_TRANSPOSE ) in FLA_Sylv_internal()
64 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Sylv_internal()
66 else if ( transb == FLA_TRANSPOSE || transb == FLA_CONJ_TRANSPOSE ) in FLA_Sylv_internal()
/dports/science/qmcpack/qmcpack-3.11.0/src/Platforms/CUDA/
H A DcuBLAS.hpp81 const cublasOperation_t& transb, in gemm() argument
94 return cublasSgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm()
99 const cublasOperation_t& transb, in gemm() argument
112 …return cublasCgemm(handle, transa, transb, m, n, k, (const cuComplex*)alpha, (const cuComplex*)A, … in gemm()
118 const cublasOperation_t& transb, in gemm() argument
131 return cublasDgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm()
136 const cublasOperation_t& transb, in gemm() argument
155 const cublasOperation_t& transb, in gemm_batched() argument
174 const cublasOperation_t& transb, in gemm_batched() argument
194 const cublasOperation_t& transb, in gemm_batched() argument
[all …]
/dports/devel/ga/ga-5.8/LinAlg/lapack+blas/
H A Dxgemm.c144 if (*transb == 'n' || *transb == 'N') in xb_sgemm()
173 if (*transb == 'n' || *transb == 'N') in xb_sgemm()
208 if (*transb == 'n' || *transb == 'N') in xb_sgemm()
237 if (*transb == 'n' || *transb == 'N') in xb_sgemm()
534 if (*transb == 'n' || *transb == 'N') in xb_dgemm()
563 if (*transb == 'n' || *transb == 'N') in xb_dgemm()
599 if (*transb == 'n' || *transb == 'N') in xb_dgemm()
628 if (*transb == 'n' || *transb == 'N') in xb_dgemm()
916 if (*transb == 'n' || *transb == 'N') in xb_zgemm()
945 if (*transb == 'n' || *transb == 'N') in xb_zgemm()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/
H A Dmkl.py22 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument
43 m = rhs.shape[0] if transb else rhs.shape[1]
48 "tvm.contrib.mkl.matmul", ins[0], ins[1], outs[0], transa, transb
55 def matmul_u8s8s32(lhs, rhs, transa=False, transb=False, **kwargs): argument
76 m = rhs.shape[0] if transb else rhs.shape[1]
81 "tvm.contrib.mkl.matmul_u8s8s32", ins[0], ins[1], outs[0], transa, transb
88 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument
110 m = rhs.shape[1] if transb else rhs.shape[2]
122 transb,
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Dgemm.cpp45 dnnl_status_t dnnl_sgemm(char transa, char transb, dim_t M, dim_t N, dim_t K, in dnnl_sgemm() argument
49 return cpu::extended_sgemm(&transb, &transa, &N, &M, &K, &alpha, B, &ldb, A, in dnnl_sgemm()
61 return cpu::gemm_s8x8s32(&transb, &transa, c2f_offsetC(&offsetc), &N, &M, in dnnl_gemm_u8s8s32()
73 return cpu::gemm_s8x8s32<int8_t>(&transb, &transa, c2f_offsetC(&offsetc), in dnnl_gemm_s8s8s32()
81 char transb, dim_t M, dim_t N, dim_t K, float alpha, in dnnl_gemm_bf16bf16f32() argument
93 dnnl_status_t dnnl_threadpool_interop_sgemm(char transa, char transb, dim_t M, in dnnl_threadpool_interop_sgemm() argument
105 dnnl_status_t dnnl_threadpool_interop_gemm_u8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_u8s8s32() argument
117 dnnl_status_t dnnl_threadpool_interop_gemm_s8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_s8s8s32() argument
123 status_t status = cpu::gemm_s8x8s32<int8_t>(&transb, &transa, 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() argument
[all …]
/dports/science/xtb/xtb-6.4.1/src/mctc/blas/
H A Dwrap3.f9057 character(len=1), intent(in), optional :: transb local
82 character(len=1), intent(in), optional :: transb local
87 if (present(transb)) then
88 trb = transb
106 character(len=1), intent(in), optional :: transb local
116 if (present(transb)) then
117 trb = transb
170 if (present(transb)) then
171 trb = transb
200 if (present(transb)) then
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cblas/
H A Dgemm_common.h72 bool transb = args[4]; in CallGemm() local
87 transb = IsInPlaceTransposed(B) ? !transb : transb; in CallGemm()
93 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallGemm()
111 bool transb = args[4]; in CallU8S8S32Gemm() local
134 transb = IsInPlaceTransposed(B) ? !transb : transb; in CallU8S8S32Gemm()
141 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallU8S8S32Gemm()
181 bool transb = args[4]; in CallBatchGemm() local
196 transb = IsInPlaceTransposed3D(B) ? !transb : transb; in CallBatchGemm()
210 op(batch_size, transb, transa, ColumnCount3D(B, transb), RowCount3D(A, transa), in CallBatchGemm()

12345678910>>...50