| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| blas_gemm.h | 30 void operator()(OpKernelContext* ctx, bool transa, bool transb, uint64 m,
41 bool transb, T alpha, typename TTypes<T>::ConstMatrix a,
48 TensorCuBlasGemm<T>()(ctx, transb, transa, n, m, k, alpha, b.data(),
49 transb ? k : n, a.data(), transa ? m : k, beta,
57 bool transb, T alpha, typename TTypes<T>::ConstMatrix a,
62 Eigen::IndexPair<Eigen::DenseIndex>(transa == false, transb == true);
|
| blas_gemm.cc | 40 bool transb, uint64 m, uint64 n, uint64 k,
55 ->ThenBlasGemm(trans[transa], trans[transb], m, n, k, alpha, a_ptr,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_matmul_op.cc | 112 // @param transb Specifies the form of op(b) used in MatMul. If transb is
134 // depending on transb parameter. Since TF uses row-major
136 // ldb = max(1,n) when transb is false, otherwise ldb = max(1,k)
144 void MklBlasGemm(bool transa, bool transb, const int m, const int n,
154 transb ? CblasTrans : CblasNoTrans, m, n, k, alpha, a, lda, b,
160 void MklBlasGemm(bool transa, bool transb, const int m, const int n,
166 transb ? CblasTrans : CblasNoTrans, m, n, k, alpha, a, lda, b,
172 void MklBlasGemm(bool transa, bool transb, const int m, const int n,
179 transb ? CblasTrans : CblasNoTrans
[all...] |
| matmul_op.h | 61 MatmulParameters(bool transa, bool transb, uint64 m, uint64 n, uint64 k,
64 transb_(transb),
71 hash_code_ = Hash64Combine(hash_code_, transb);
|
| cuda_solvers.cc | 327 cublasOperation_t transb, int m, int n,
334 TF_RETURN_IF_CUBLAS_ERROR(solver(cublas_handle, transa, transb, m, n,
346 cublasOperation_t transa, cublasOperation_t transb, int m, int n, \
352 transb, m, n, alpha, A, lda, beta, B, ldb, C, ldc); \
[all...] |
| cuda_solvers.h | 198 Status Geam(cublasOperation_t transa, cublasOperation_t transb, int m, int n,
|
| /external/eigen/Eigen/src/Core/products/ |
| GeneralMatrixMatrix_BLAS.h | 68 char transa, transb; \
76 transb = (RhsStorageOrder==RowMajor) ? ((ConjugateRhs) ? 'C' : 'T') : 'N'; \
103 BLASPREFIX##gemm_(&transa, &transb, &m, &n, &k, &numext::real_ref(alpha), (const BLASTYPE*)a, &lda, (const BLASTYPE*)b, &ldb, &numext::real_ref(beta), (BLASTYPE*)res, &ldc); \
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_blas.h | 113 blas::Transpose transb, uint64 m, uint64 n, uint64 k, T alpha,
127 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
137 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
|
| cuda_blas.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| blas.h | [all...] |
| stream.cc | [all...] |
| stream.h | [all...] |
| /external/cblas/testing/ |
| c_dblas3.c | 20 enum CBLAS_TRANSPOSE transa, transb; local
23 get_transpose_type(transpb, &transb);
40 if (transb == CblasNoTrans) {
60 cblas_dgemm( CblasRowMajor, transa, transb, *m, *n, *k, *alpha, A, LDA,
70 cblas_dgemm( CblasColMajor, transa, transb, *m, *n, *k, *alpha, a, *lda,
73 cblas_dgemm( UNDEFINED, transa, transb, *m, *n, *k, *alpha, a, *lda,
|
| c_sblas3.c | 18 enum CBLAS_TRANSPOSE transa, transb; local
21 get_transpose_type(transpb, &transb);
38 if (transb == CblasNoTrans) {
57 cblas_sgemm( CblasRowMajor, transa, transb, *m, *n, *k, *alpha, A, LDA,
67 cblas_sgemm( CblasColMajor, transa, transb, *m, *n, *k, *alpha, a, *lda,
70 cblas_sgemm( UNDEFINED, transa, transb, *m, *n, *k, *alpha, a, *lda,
|
| c_cblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; local
24 get_transpose_type(transpb, &transb);
46 if (transb == CblasNoTrans) {
72 cblas_cgemm( CblasRowMajor, transa, transb, *m, *n, *k, alpha, A, LDA,
84 cblas_cgemm( CblasColMajor, transa, transb, *m, *n, *k, alpha, a, *lda,
87 cblas_cgemm( UNDEFINED, transa, transb, *m, *n, *k, alpha, a, *lda,
|
| c_zblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; local
24 get_transpose_type(transpb, &transb);
46 if (transb == CblasNoTrans) {
72 cblas_zgemm( CblasRowMajor, transa, transb, *m, *n, *k, alpha, A, LDA,
84 cblas_zgemm( CblasColMajor, transa, transb, *m, *n, *k, alpha, a, *lda,
87 cblas_zgemm( UNDEFINED, transa, transb, *m, *n, *k, alpha, a, *lda,
|
| /external/opencv/cxcore/src/ |
| _cxipp.h | [all...] |
| cxmatmul.cpp | 1042 const char* transb = flags & CV_GEMM_B_T ? "t" : "n"; local
[all...] |
