Home
last modified time | relevance | path

Searched defs:transa (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc389 port::Status ROCMBlas::DoBlasGemm(Stream *stream, blas::Transpose transa, in DoBlasGemm()
555 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfiling()
566 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfiling()
576 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfiling()
587 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfiling()
599 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfiling()
622 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithProfilingImpl()
630 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmWithAlgorithm()
641 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasGemmStridedBatchedWithAlgorithm()
766 FuncT rocblas_func, Stream *stream, blas::Transpose transa, in DoBlasGemmBatchedInternal()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc608 port::Status CUDABlas::DoBlasGemm(Stream *stream, blas::Transpose transa, in DoBlasInternalImpl()
790 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
801 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
811 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
822 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
834 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
879 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
1016 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
1045 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64_t m, in DoBlasInternalImpl()
1188 FuncT cublas_func, Stream *stream, blas::Transpose transa, in DoBlasInternalImpl()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.h754 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm()
768 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm()
778 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm()
816 port::Status ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm()
866 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithAlgorithm()
881 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithAlgorithm()
919 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmStridedBatchedWithAlgorithm()
1034 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmStridedBatched()
Dstream.cc1452 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling()
1472 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling()
1491 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling()
1511 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling()
1533 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmWithProfiling()
1555 blas::Transpose transa, blas::Diagonal diag, in ThenBlasTrsm()
1571 blas::Transpose transa, blas::Diagonal diag, in ThenBlasTrsm()
1587 blas::Transpose transa, blas::Diagonal diag, in ThenBlasTrsm()
1605 blas::Transpose transa, blas::Diagonal diag, in ThenBlasTrsm()
1623 blas::Transpose transa, blas::Diagonal diag, in ThenBlasTrsmBatched()
[all …]
/external/cblas/examples/
Dcblas_example1.c10 enum CBLAS_TRANSPOSE transa; in main() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op.cc146 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm()
179 void MklBlasGemm(OpKernelContext* ctx, bool transa, bool transb, const int m, in MklBlasGemm()
/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
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.cc39 void TensorCuBlasGemm<T>::operator()(OpKernelContext* ctx, bool transa, in operator ()()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.h65 MatmulParameters(bool transa, bool transb, uint64 m, uint64 n, uint64 k, in MatmulParameters()
/external/cblas/testing/
Dc_dblas3.c20 enum CBLAS_TRANSPOSE transa, transb; in F77_dgemm() local
Dc_sblas3.c18 enum CBLAS_TRANSPOSE transa, transb; in F77_sgemm() local
Dc_cblas3.c21 enum CBLAS_TRANSPOSE transa, transb; in F77_cgemm() local
Dc_zblas3.c21 enum CBLAS_TRANSPOSE transa, transb; in F77_zgemm() local
/external/tensorflow/tensorflow/core/util/
Drocm_solvers.cc810 rocblas_handle rocm_blas_handle, rocblas_operation transa, in GeamImpl()
Dcuda_solvers.cc346 cublasOperation_t transa, in GeamImpl()
/external/igt-gpu-tools/tools/
Dintel_reg_decode.c1577 const char *transa, *transb; in DEBUGSTRING() local