Home
last modified time | relevance | path

Searched defs:transb_ (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUBlas.cpp91 const bool transb_ = transb != TransposeType::NoTranspose; in use_blas_gemm() local
142 CBLAS_TRANSPOSE transb_ = to_apple_accelerate_transpose(transb); in gemm() local
152 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
190 CBLAS_TRANSPOSE transb_ = to_apple_accelerate_transpose(transb); in gemm() local
200 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
233 CBLAS_TRANSPOSE transb_ = to_apple_accelerate_transpose(transb); in gemm() local
243 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
276 CBLAS_TRANSPOSE transb_ = to_apple_accelerate_transpose(transb); in gemm() local
286 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
316 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
[all …]
/aosp_15_r20/external/executorch/kernels/optimized/blas/
H A DCPUBlas.cpp90 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
130 char transa_ = to_blas(transa), transb_ = to_blas(transb); in gemm() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op.h109 bool transb_; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBlas.cpp219 bool transb_ = ((args.transb != 'n') && (args.transb != 'N')); in launchTunableGemmAndBias() local
1108 bool transb_ = ((args.transb != 'n') && (args.transb != 'N')); in _scaled_mm_out_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp145 bool transb_ = ((transb != 'n') && (transb != 'N')); in _cublasAdjustLdLevel3() local
695 bool transb_ = ((transb != 'n') && (transb != 'N')); in bgemm_tunable() local
1091 bool transb_ = ((transb != 'n') && (transb != 'N')); in gemm_tunable() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDABlas.cpp266 int transb_ = ((transb == 't') || (transb == 'T')); in adjustLd() local