| /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()
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()
353 void Zcsrmm2(char transa, char transb, int64_t m, int64_t n, int64_t k, int64_t nnz, const complex_… in Zcsrmm2()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 389 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas.cc | 608 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 …]
|
| /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()
464 const char transa = self.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/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()
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 …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | BlasKernel.cpp | 166 TransposeType transa, in gemm_transa_()
305 TransposeType transa, TransposeType transb, in gemm_transab_()
383 TransposeType transa, in gemm_transa_()
415 TransposeType transa, in gemm_transa_()
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 | 141 char transa; member
222 char transa; member
306 char transa; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | qr_op_impl.h | 218 cublasOperation_t transa = CUBLAS_OP_T; in ComputeAsync() local
222 rocblas_operation transa = rocblas_operation_transpose; in ComputeAsync() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 136 char transa, in _cublasAdjustLdLevel3()
1221 cublasOperation_t transa = transpose_mat1 ? CUBLAS_OP_T : CUBLAS_OP_N; in gemm_and_bias() local
1392 char transa, in scaled_gemm()
1587 cublasOperation_t transa = transpose_mat1 ? CUBLAS_OP_T : CUBLAS_OP_N; in int8_gemm() local
|
| /aosp_15_r20/external/cblas/examples/ |
| H A D | cblas_example1.c | 10 enum CBLAS_TRANSPOSE transa; in main() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | blas_gemm.cc | 39 void TensorCuBlasGemm<T>::operator()(OpKernelContext* ctx, bool transa, 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
562 const auto transa = transpose_batch1 ? batch1_->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 | 810 rocblas_handle rocm_blas_handle, rocblas_operation transa, 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
|