Searched defs:transA (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_sparse.cc | 191 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl()
262 hipsparseOperation_t transA, int m, int n, in CsrmvImpl()
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()
|
| D | cuda_sparse.cc | 394 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, in CsrmmImpl()
473 cusparseOperation_t transA, int m, int n, int nnz, const Scalar* alpha_host, in CsrmvImpl()
514 cusparseOperation_t transA, int m, int n, in CsrmvExImpl()
553 cusparseOperation_t transA, int m, int n, int nnz, in SpMVImpl()
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()
|
| /external/eigen/bench/ |
| D | bench_gemm.cpp | 75 const char transA = trans; variable
77 const char transA = notrans; variable
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 831 const gpusparseOperation_t transA = CUSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
833 const gpusparseOperation_t transA = HIPSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 1398 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1406 char transA, const bfloat16* A, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1416 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
1426 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_compute_generic_thread()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 1993 int transA = static_cast<int>(ReadOptionalNodeUint32Attribute(node, "transA", 0)); in ParseGemm() local
|