Searched defs:transB (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | mat_mul_op.cc | 841 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
879 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
919 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
930 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | rocm_sparse.cc | 191 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl()
295 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmNnz()
313 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmImpl()
|
| H A D | cuda_sparse.cc | 394 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, in CsrmmImpl()
764 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmNnz()
782 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmImpl()
|
| /aosp_15_r20/external/eigen/bench/ |
| H A D | bench_gemm.cpp | 81 const char transB = trans; variable
83 const char transB = notrans; variable
|
| /aosp_15_r20/cts/tests/tests/rsblas/src/android/renderscript/cts/ |
| H A D | IntrinsicBLAS.java | 4993 private void xGEMM_API_test(int transA, int transB, ArrayList<Allocation> mMatrix) { in xGEMM_API_test()
5066 int transB = ScriptIntrinsicBLAS.NO_TRANSPOSE; in test_L3_SGEMM_Correctness() local
5109 int transB = ScriptIntrinsicBLAS.NO_TRANSPOSE; in test_L3_DGEMM_Correctness() local
5151 int transB = ScriptIntrinsicBLAS.NO_TRANSPOSE; in test_L3_CGEMM_Correctness() local
5194 int transB = ScriptIntrinsicBLAS.NO_TRANSPOSE; in test_L3_ZGEMM_Correctness() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_matmul_op.cc | 1416 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
1427 char transB, const float* alpha, libxsmm_CSR_sparseslice* A_sparse, in wrapper_libxsmm_spmdm_compute_generic_thread()
|
| /aosp_15_r20/frameworks/rs/ |
| H A D | rsDefines.h | 456 RsBlasTranspose transB; member
|
| /aosp_15_r20/external/armnn/src/armnnOnnxParser/ |
| H A D | OnnxParser.cpp | 1994 int transB = static_cast<int>(ReadOptionalNodeUint32Attribute(node, "transB", 0)); in ParseGemm() local
|