| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDABlas.cpp | 100 char transa, char transb, in _csrmm2()
202 char transa, char transb, in csrmm2()
211 char transa, char transb, in csrmm2()
220 char transa, char transb, in csrmm2()
229 char transa, char transb, in csrmm2()
246 char transa, char transb, in csrmm2()
264 void adjustLd(char transb, int64_t m, int64_t n, int64_t k, int64_t *ldb, int64_t *ldc) in adjustLd()
283 void Scsrmm2(char transa, char transb, int64_t m, int64_t n, int64_t k, int64_t nnz, const float *a… in Scsrmm2()
305 void Dcsrmm2(char transa, char transb, int64_t m, int64_t n, int64_t k, int64_t nnz, const double *… in Dcsrmm2()
330 void Ccsrmm2(char transa, char transb, int64_t m, int64_t n, int64_t k, int64_t nnz, const complex_… in Ccsrmm2()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | CPUBlas.cpp | 59 TransposeType transa, TransposeType transb, in normalize_last_dims()
87 TransposeType transa, TransposeType transb, in use_blas_gemm()
128 TransposeType transa, TransposeType transb, in gemm()
171 TransposeType transa, TransposeType transb, in gemm()
219 TransposeType transa, TransposeType transb, in gemm()
262 TransposeType transa, TransposeType transb, in gemm()
305 TransposeType transa, TransposeType transb, in gemm()
345 TransposeType transa, TransposeType transb, in gemm()
364 TransposeType transa, TransposeType transb, in gemm()
415 TransposeType transa, TransposeType transb, in gemm()
[all …]
|
| H A D | CPUBlas.h | 34 TransposeType transa, TransposeType transb, in gemm()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Matmul.cpp | 36 TransposeType transa, TransposeType transb, in mkldnn_bf16_gemm()
47 TransposeType transa, TransposeType transb, in mkldnn_fp16_gemm()
57 TransposeType transa, TransposeType transb, in mkldnn_bf32_gemm()
119 TransposeType transa, TransposeType transb, in mkldnn_gemm()
195 TransposeType transa, TransposeType transb, in mkldnn_bf16_gemm()
206 TransposeType transa, TransposeType transb, in mkldnn_fp16_gemm()
217 TransposeType transa, TransposeType transb, in mkldnn_bf32_gemm()
465 const char transb = mat2.strides()[1] == 1 ? 'N' : 'T'; in _mkldnn_gemm_i8i8i32_with_blas() local
|
| /aosp_15_r20/external/executorch/kernels/optimized/blas/ |
| H A D | CPUBlas.cpp | 49 TransposeType transa, TransposeType transb, in normalize_last_dims()
76 TransposeType transa, TransposeType transb, in gemm()
116 TransposeType transa, TransposeType transb, in gemm()
156 TransposeType transa, TransposeType transb, in gemm()
179 TransposeType transa, TransposeType transb, in gemm()
|
| H A D | CPUBlas.h | 49 TransposeType transa, TransposeType transb, in gemm_impl()
120 TransposeType transa, TransposeType transb, in gemm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 390 blas::Transpose transb, uint64_t m, uint64 n, 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()
767 blas::Transpose transb, uint64_t m, uint64 n, uint64 k, T alpha, in DoBlasGemmBatchedInternal()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.h | 754 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()
|
| H A D | stream.cc | 1452 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()
1695 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmBatched()
1709 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmBatchedWithScratch()
1736 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmBatched()
1749 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmBatchedWithScratch()
1773 blas::Transpose transa, blas::Transpose transb, uint64_t m, uint64 n, in ThenBlasGemmBatched()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas.cc | 609 blas::Transpose transb, uint64_t m, uint64 n, 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()
1189 blas::Transpose transb, uint64_t m, uint64 n, uint64 k, Scalar alpha, in DoBlasInternalImpl()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | BlasKernel.cpp | 194 TransposeType transb, in gemm_transb_impl()
227 TransposeType transb, in gemm_transb_()
249 TransposeType transb, in gemm_transb_()
305 TransposeType transa, TransposeType transb, in gemm_transab_()
445 TransposeType transa, TransposeType transb, in gemm_core_()
481 TransposeType transa, TransposeType transb, in cpublas_gemm_impl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_matmul_op.cc | 146 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/ |
| H A D | GemmCommon.h | 142 char transb; member
223 char transb; member
307 char transb; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | qr_op_impl.h | 219 cublasOperation_t transb = CUBLAS_OP_N; in ComputeAsync() local
223 rocblas_operation transb = rocblas_operation_none; in ComputeAsync() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 137 char transb, in _cublasAdjustLdLevel3()
1223 cublasOperation_t transb = transpose_mat2 ? CUBLAS_OP_T : CUBLAS_OP_N; in gemm_and_bias() local
1393 char transb, in scaled_gemm()
1589 cublasOperation_t transb = transpose_mat2 ? CUBLAS_OP_T : CUBLAS_OP_N; in int8_gemm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | blas_gemm.cc | 40 bool transb, uint64 m, uint64 n, uint64 k, in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_op.h | 65 MatmulParameters(bool transa, bool transb, uint64 m, uint64 n, uint64 k, in MatmulParameters()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Blas.cpp | 120 char transa, transb; member
563 const auto transb = transpose_batch2 ? batch2_->is_conj() ? 'c' : 't' : 'n'; in baddbmm_out_cuda_impl() local
|
| /aosp_15_r20/external/cblas/testing/ |
| H A D | c_sblas3.c | 18 enum CBLAS_TRANSPOSE transa, transb; in F77_sgemm() local
|
| H A D | c_dblas3.c | 20 enum CBLAS_TRANSPOSE transa, transb; in F77_dgemm() local
|
| H A D | c_zblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; in F77_zgemm() local
|
| H A D | c_cblas3.c | 21 enum CBLAS_TRANSPOSE transa, transb; in F77_cgemm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | rocm_solvers.cc | 811 rocblas_operation transb, int m, int n, const Scalar* alpha, in GeamImpl()
|
| H A D | cuda_solvers.cc | 347 cublasOperation_t transb, int m, int n, in GeamImpl()
|
| /aosp_15_r20/external/igt-gpu-tools/tools/ |
| H A D | intel_reg_decode.c | 1577 const char *transa, *transb; in DEBUGSTRING() local
|