| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_matmul_op.cc | 149 void MklBlasGemm(bool transa, bool transb, const int m, const int n, in MklBlasGemm() argument
161 int index_transb = transb ? 1 : 0; in MklBlasGemm()
170 transb ? CblasTrans : CblasNoTrans, m, n, k, alpha, a, lda, b, in MklBlasGemm()
180 void MklBlasGemm(bool transa, bool transb, const int m, const int n, in MklBlasGemm() argument
186 transb ? CblasTrans : CblasNoTrans, m, n, k, alpha, a, lda, b, in MklBlasGemm()
192 void MklBlasGemm(bool transa, bool transb, const int m, const int n, in MklBlasGemm() argument
199 transb ? CblasTrans : CblasNoTrans, m, n, k, &alpha, in MklBlasGemm()
208 void MklBlasGemm(bool transa, bool transb, const int m, const int n, in MklBlasGemm() argument
215 transb ? CblasTrans : CblasNoTrans, m, n, k, &alpha, in MklBlasGemm()
|
| D | matmul_op.h | 65 MatmulParameters(bool transa, bool transb, uint64 m, uint64 n, uint64 k, in MatmulParameters() argument
68 transb_(transb), in MatmulParameters()
75 hash_code_ = Hash64Combine(hash_code_, transb); in MatmulParameters()
|
| D | cuda_solvers.cc | 327 cublasOperation_t transb, int m, int n, in GeamImpl() argument
334 TF_RETURN_IF_CUBLAS_ERROR(solver(cublas_handle, transa, transb, m, n, in GeamImpl()
346 cublasOperation_t transa, cublasOperation_t transb, int m, int n, \
352 transb, m, n, alpha, A, lda, beta, B, ldb, C, ldc); \
|
| D | cuda_solvers.h | 201 Status Geam(cublasOperation_t transa, cublasOperation_t transb, int m, int n,
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | blas_gemm.h | 34 void operator()(OpKernelContext* ctx, bool transa, bool transb, uint64 m,
55 bool transb, typename gemm_compute_type<T>::type alpha,
64 TensorCuBlasGemm<T>()(ctx, transb, transa, n, m, k, alpha, b.data(),
65 transb ? k : n, a.data(), transa ? m : k, beta,
73 bool transb, typename gemm_compute_type<T>::type alpha,
80 Eigen::IndexPair<Eigen::DenseIndex>(transa == false, transb == true);
|
| D | blas_gemm.cc | 40 bool transb, uint64 m, uint64 n, uint64 k, in operator ()() argument
55 ->ThenBlasGemm(trans[transa], trans[transb], m, n, k, alpha, a_ptr, in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 1562 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1571 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1584 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1611 CUDABlasTranspose(transb), m, n, k, &alpha, GpuMemory(a), in DoBlasGemm()
1623 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1631 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1644 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1656 CUDABlasTranspose(transa), CUDABlasTranspose(transb), m, in DoBlasGemm()
1662 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1667 CUDABlasTranspose(transa), CUDABlasTranspose(transb), m, in DoBlasGemm()
[all …]
|
| D | cuda_blas.h | 112 blas::Transpose transb, uint64 m, uint64 n, uint64 k, Scalar alpha,
121 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
131 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
|
| D | cublas_9_0.inc | 2031 cublasOperation_t transb,
2046 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2051 cublasOperation_t transb,
2066 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2071 cublasOperation_t transb,
2086 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2091 cublasOperation_t transb,
2106 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2110 … cublasOperation_t transa, cublasOperation_t transb,
2126 …return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B, Btype, ldb, beta, C, Cty…
[all …]
|
| D | cublas_10_0.inc | 2052 cublasOperation_t transb,
2067 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2072 cublasOperation_t transb,
2087 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2092 cublasOperation_t transb,
2107 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2112 cublasOperation_t transb,
2127 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
2131 … cublasOperation_t transa, cublasOperation_t transb,
2147 …return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B, Btype, ldb, beta, C, Cty…
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1461 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1469 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1482 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1497 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
1506 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1514 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1527 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1540 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
1545 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1551 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
[all …]
|
| D | rocm_blas.h | 105 blas::Transpose transb, uint64 m, uint64 n, uint64 k, T alpha,
119 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
129 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
|
| /external/eigen/Eigen/src/Core/products/ |
| D | 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…
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 955 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
961 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
966 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
971 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
978 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
986 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
992 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
997 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
1003 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
1010 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
[all …]
|
| D | stream.cc | 3498 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3504 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3512 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3516 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3521 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3528 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3532 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3537 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3544 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3548 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
[all …]
|
| D | stream.h | 1250 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1256 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1261 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1266 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1274 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1285 blas::Transpose transb, uint64 m, uint64 n,
1293 blas::Transpose transb, uint64 m, uint64 n,
1300 blas::Transpose transb, uint64 m, uint64 n,
1308 blas::Transpose transa, blas::Transpose transb, uint64 m, uint64 n,
1315 blas::Transpose transa, blas::Transpose transb, uint64 m, uint64 n,
[all …]
|
| /external/cblas/testing/ |
| D | c_sblas3.c | 18 enum CBLAS_TRANSPOSE transa, transb; in F77_sgemm() local
21 get_transpose_type(transpb, &transb); in F77_sgemm()
38 if (transb == CblasNoTrans) { in F77_sgemm()
57 cblas_sgemm( CblasRowMajor, transa, transb, *m, *n, *k, *alpha, A, LDA, in F77_sgemm()
67 cblas_sgemm( CblasColMajor, transa, transb, *m, *n, *k, *alpha, a, *lda, in F77_sgemm()
70 cblas_sgemm( UNDEFINED, transa, transb, *m, *n, *k, *alpha, a, *lda, in F77_sgemm()
|
| D | c_dblas3.c | 20 enum CBLAS_TRANSPOSE transa, transb; in F77_dgemm() local
23 get_transpose_type(transpb, &transb); in F77_dgemm()
40 if (transb == CblasNoTrans) { in F77_dgemm()
60 cblas_dgemm( CblasRowMajor, transa, transb, *m, *n, *k, *alpha, A, LDA, in F77_dgemm()
70 cblas_dgemm( CblasColMajor, transa, transb, *m, *n, *k, *alpha, a, *lda, in F77_dgemm()
73 cblas_dgemm( UNDEFINED, transa, transb, *m, *n, *k, *alpha, a, *lda, in F77_dgemm()
|
| D | c_zblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; in F77_zgemm() local
24 get_transpose_type(transpb, &transb); in F77_zgemm()
46 if (transb == CblasNoTrans) { in F77_zgemm()
72 cblas_zgemm( CblasRowMajor, transa, transb, *m, *n, *k, alpha, A, LDA, in F77_zgemm()
84 cblas_zgemm( CblasColMajor, transa, transb, *m, *n, *k, alpha, a, *lda, in F77_zgemm()
87 cblas_zgemm( UNDEFINED, transa, transb, *m, *n, *k, alpha, a, *lda, in F77_zgemm()
|
| D | c_cblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; in F77_cgemm() local
24 get_transpose_type(transpb, &transb); in F77_cgemm()
46 if (transb == CblasNoTrans) { in F77_cgemm()
72 cblas_cgemm( CblasRowMajor, transa, transb, *m, *n, *k, alpha, A, LDA, in F77_cgemm()
84 cblas_cgemm( CblasColMajor, transa, transb, *m, *n, *k, alpha, a, *lda, in F77_cgemm()
87 cblas_cgemm( UNDEFINED, transa, transb, *m, *n, *k, alpha, a, *lda, in F77_cgemm()
|