Searched defs:transB (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| 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
|
| /external/eigen/bench/ |
| D | bench_gemm.cpp | 81 const char transB = trans; variable
83 const char transB = notrans; variable
|
| /external/tensorflow/tensorflow/core/util/ |
| 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()
|
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 1994 int transB = static_cast<int>(ReadOptionalNodeUint32Attribute(node, "transB", 0)); in ParseGemm() local
|