Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.h34 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);
Dblas_gemm.cc40 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/core/kernels/mkl/
Dmkl_matmul_op.cc146 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm() argument
156 char char_transb = transb ? 'T' : 'N'; in MklBlasGemm()
171 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm() argument
178 const int index_transb = transb ? 1 : 0; in MklBlasGemm()
Dmkl_matmul_ops_common.h695 void dnnl_gemm(char transa, char transb, int64_t m, int64_t n, int64_t k,
706 dims b_strides = tolower(transb) == 'n' ? dims{ldb, 1} : dims{1, ldb};
/external/eigen/Eigen/src/Core/products/
DGeneralMatrixMatrix_BLAS.h68 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/core/kernels/
Dmatmul_op.h65 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()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc1740 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasInternalImpl() argument
1749 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasInternalImpl()
1762 if (transb == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
1783 CUDABlasTranspose(transb), m, n, k, &alpha, GpuMemory(a), in DoBlasInternalImpl()
1795 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasInternalImpl() argument
1803 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasInternalImpl()
1816 if (transb == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
1846 CUDABlasTranspose(transb), m, n, k, &alpha, GpuMemory(a), lda, in DoBlasInternalImpl()
1851 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasInternalImpl() argument
1856 CUDABlasTranspose(transa), CUDABlasTranspose(transb), m, in DoBlasInternalImpl()
[all …]
Dcuda_blas.h109 blas::Transpose transb, uint64 m, uint64 n, uint64 k, Scalar alpha,
118 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
128 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
Dcublas_9_0.inc1820 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1831 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1836 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1847 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1852 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1863 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1868 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1879 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1884 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1894 return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B,
[all …]
Dcublas_11_0.inc1970 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1981 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1986 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1997 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2002 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2013 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2018 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2029 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2034 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2044 return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B,
[all …]
Dcublas_10_1.inc1970 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1981 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1986 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1997 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2002 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2013 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2018 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2029 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2034 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2044 return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B,
[all …]
Dcublas_10_0.inc1845 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1856 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1861 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1872 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1877 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1888 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1893 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1904 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1909 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1919 return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B,
[all …]
Dcublas_10_2.inc1970 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1981 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
1986 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
1997 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2002 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2013 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2018 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2029 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2034 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2044 return func_ptr(handle, transa, transb, m, n, k, alpha, A, Atype, lda, B,
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1587 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1596 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1609 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1624 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
1633 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1642 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
1655 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1668 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
1673 blas::Transpose transb, uint64 m, uint64 n, uint64 k, in DoBlasGemm() argument
1680 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
[all …]
Drocm_blas.h156 blas::Transpose transb, uint64 m, uint64 n, uint64 k, T alpha,
170 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
180 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
/external/tensorflow/tensorflow/stream_executor/
Dblas.h221 Transpose transb; member
1024 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
1030 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
1035 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
1040 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
1047 blas::Transpose transb, uint64 m, uint64 n, uint64 k,
1055 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
1061 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
1066 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
1072 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
[all …]
Dstream.cc3138 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3144 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3153 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3157 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3162 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3170 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3174 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
3179 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemm()
3187 return impl(this, &blas::BlasSupport::DoBlasGemm, transa, transb, m, n, k, in ThenBlasGemm()
3191 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
[all …]
Dstream.h1273 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1279 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1284 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1289 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1297 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1308 blas::Transpose transb, uint64 m, uint64 n,
1316 blas::Transpose transb, uint64 m, uint64 n,
1323 blas::Transpose transb, uint64 m, uint64 n,
1331 blas::Transpose transa, blas::Transpose transb, uint64 m, uint64 n,
1338 blas::Transpose transa, blas::Transpose transb, uint64 m, uint64 n,
[all …]
/external/cblas/testing/
Dc_dblas3.c20 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()
Dc_sblas3.c18 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()
Dc_cblas3.c21 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()
Dc_zblas3.c21 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()
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc348 cublasOperation_t transb, int m, int n, in GeamImpl() argument
355 TF_RETURN_IF_CUBLAS_ERROR(solver(cublas_handle, transa, transb, m, n, in GeamImpl()
367 cublasOperation_t transa, cublasOperation_t transb, int m, int n, \
373 transb, m, n, alpha, A, lda, beta, B, ldb, C, ldc); \
Dcuda_solvers.h208 Status Geam(cublasOperation_t transa, cublasOperation_t transb, int m, int n,
/external/ruy/ruy/
Dtest.h1137 static void Run(char* transa, char* transb, lapack::integer* m,
1143 dgemm_(transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);
1149 static void Run(char* transa, char* transb, lapack::integer* m,
1154 sgemm_(transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);
1216 char transb = transposed_rhs ? 'T' : 'N';
1228 GenericBlasGemm<Scalar>::Run(&transa, &transb, &m, &n, &k, &alpha, a, &lda, b,

12