Home
last modified time | relevance | path

Searched refs:transa (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc389 port::Status ROCMBlas::DoBlasGemm(Stream *stream, blas::Transpose transa, in DoBlasGemm() argument
401 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasGemm()
404 if (transa == blas::Transpose::kNoTranspose) { in DoBlasGemm()
435 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), in DoBlasGemm()
447 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
458 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), (rocblas_int)m, in DoBlasGemm()
467 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
476 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
489 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
501 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
[all …]
Drocm_blas.h167 FuncT rocblas_func, Stream *stream, blas::Transpose transa,
176 bool DoBlasGemmWithProfilingImpl(Stream *stream, blas::Transpose transa,
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dblas.h347 virtual port::Status DoBlasGemm(Stream *stream, blas::Transpose transa,
356 Stream *stream, blas::Transpose transa, blas::Transpose transb,
363 Stream *stream, blas::Transpose transa, blas::Transpose transb,
369 Stream *stream, blas::Transpose transa, blas::Transpose transb,
375 Stream *stream, blas::Transpose transa, blas::Transpose transb,
382 Stream *stream, blas::Transpose transa, blas::Transpose transb,
405 Stream *stream, blas::Transpose transa, blas::Transpose transb,
414 Stream *stream, blas::Transpose transa, blas::Transpose transb,
427 Stream *stream, blas::Transpose transa, blas::Transpose transb,
436 Stream *stream, blas::Transpose transa, blas::Transpose transb,
[all …]
Dstream.h754 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
762 return ThenBlasGemm(transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, in ThenBlasGemm()
768 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
773 return ThenBlasGemm(transa, transb, m, n, k, a, lda, b, ldb, c, ldc, in ThenBlasGemm()
778 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
809 return blas->DoBlasGemm(this, transa, transb, m, n, k, in ThenBlasGemm()
816 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() argument
822 return ThenBlasGemm(transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, in ThenBlasGemm()
826 Stream &ThenBlasGemmWithProfiling(blas::Transpose transa,
834 Stream &ThenBlasGemmWithProfiling(blas::Transpose transa,
[all …]
Dstream.cc1452 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling() argument
1457 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemmWithProfiling()
1466 return impl(this, &blas::BlasSupport::DoBlasGemmWithProfiling, transa, transb, in ThenBlasGemmWithProfiling()
1472 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling() argument
1476 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemmWithProfiling()
1485 return impl(this, &blas::BlasSupport::DoBlasGemmWithProfiling, transa, transb, in ThenBlasGemmWithProfiling()
1491 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling() argument
1496 VLOG_CALL(PARAM(transa), PARAM(transb), PARAM(m), PARAM(n), PARAM(k), in ThenBlasGemmWithProfiling()
1505 return impl(this, &blas::BlasSupport::DoBlasGemmWithProfiling, transa, transb, in ThenBlasGemmWithProfiling()
1511 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc608 port::Status CUDABlas::DoBlasGemm(Stream *stream, blas::Transpose transa, in DoBlasInternalImpl() argument
643 if (transa == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
671 static_cast<int>(transa), static_cast<int>(transb), m, n, k, alpha, in DoBlasInternalImpl()
684 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
693 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
702 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
711 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
724 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
736 AsCublasOperation(transa), AsCublasOperation(transb), m, n, k, in DoBlasInternalImpl()
790 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl() argument
[all …]
Dcuda_blas.h110 FuncT cublas_func, Stream *stream, blas::Transpose transa,
119 bool DoBlasGemmWithProfilingImpl(Stream *stream, blas::Transpose transa,
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_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 …]
Dcublas_11_0.inc2144 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2155 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2160 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2171 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2176 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2187 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2192 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2203 return func_ptr(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta,
2208 cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb,
2218 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 …]
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.h34 void operator()(OpKernelContext* ctx, bool transa, bool transb, uint64 m,
54 static void compute(OpKernelContext* ctx, const Device& d, bool transa,
62 int64_t k = transa ? a.dimensions()[0] : a.dimensions()[1];
64 TensorCuBlasGemm<T>()(ctx, transb, transa, n, m, k, alpha, b.data(),
65 transb ? k : n, a.data(), transa ? m : k, beta,
72 static void compute(OpKernelContext* ctx, const Device& d, bool transa,
80 Eigen::IndexPair<Eigen::DenseIndex>(transa == false, transb == true);
Dblas_gemm.cc39 void TensorCuBlasGemm<T>::operator()(OpKernelContext* ctx, bool transa, in operator ()() argument
54 trans[transa], trans[transb], m, n, k, static_cast<T>(alpha), in operator ()()
/external/eigen/Eigen/src/Core/products/
DTriangularSolverMatrix_BLAS.h59 char side = 'L', uplo, diag='N', transa; \
66 transa = (TriStorageOrder==RowMajor) ? ((Conjugate) ? 'C' : 'T') : 'N'; \
85 …BLASFUNC(&side, &uplo, &transa, &diag, &m, &n, (const BLASTYPE*)&numext::real_ref(alpha), (const B…
120 char side = 'R', uplo, diag='N', transa; \
127 transa = (TriStorageOrder==RowMajor) ? ((Conjugate) ? 'C' : 'T') : 'N'; \
146 …BLASFUNC(&side, &uplo, &transa, &diag, &m, &n, (const BLASTYPE*)&numext::real_ref(alpha), (const B…
DGeneralMatrixMatrix_BLAS.h70 char transa, transb; \
77 transa = (LhsStorageOrder==RowMajor) ? ((ConjugateLhs) ? 'C' : 'T') : 'N'; \
105 …BLASFUNC(&transa, &transb, &m, &n, &k, (const BLASTYPE*)&numext::real_ref(alpha), (const BLASTYPE*…
DTriangularMatrixMatrix_BLAS.h136 char side = 'L', transa, uplo, diag = 'N'; \
146 transa = (LhsStorageOrder==RowMajor) ? ((ConjugateLhs) ? 'C' : 'T') : 'N'; \
177 …BLASFUNC(&side, &uplo, &transa, &diag, &m, &n, (const BLASTYPE*)&numext::real_ref(alpha), (const B…
253 char side = 'R', transa, uplo, diag = 'N'; \
263 transa = (RhsStorageOrder==RowMajor) ? ((ConjugateRhs) ? 'C' : 'T') : 'N'; \
294 …BLASFUNC(&side, &uplo, &transa, &diag, &m, &n, (const BLASTYPE*)&numext::real_ref(alpha), (const B…
/external/cblas/examples/
Dcblas_example1.c10 enum CBLAS_TRANSPOSE transa; in main() local
17 transa = CblasNoTrans; in main()
60 cblas_dgemv( order, transa, m, n, alpha, a, lda, x, incx, beta, in main()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op.cc146 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm() argument
155 char char_transa = transa ? 'T' : 'N'; in MklBlasGemm()
179 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm() argument
185 const int index_transa = transa ? 1 : 0; in MklBlasGemm()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.h65 MatmulParameters(bool transa, bool transb, uint64 m, uint64 n, uint64 k, in MatmulParameters() argument
67 : transa_(transa), in MatmulParameters()
74 hash_code_ = transa; in MatmulParameters()
/external/cblas/testing/
Dc_dblas3.c20 enum CBLAS_TRANSPOSE transa, transb; in F77_dgemm() local
22 get_transpose_type(transpa, &transa); in F77_dgemm()
26 if (transa == 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
20 get_transpose_type(transpa, &transa); in F77_sgemm()
24 if (transa == 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
23 get_transpose_type(transpa, &transa); in F77_cgemm()
27 if (transa == 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
23 get_transpose_type(transpa, &transa); in F77_zgemm()
27 if (transa == 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/kernels/linalg/
Dqr_op_impl.h218 cublasOperation_t transa = CUBLAS_OP_T; in ComputeAsync() local
222 rocblas_operation transa = rocblas_operation_transpose; in ComputeAsync() local
240 solver->Geam(transa, transb, n, full_matrices_ ? m : min_size, in ComputeAsync()

12