Searched defs:transB (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_sparse.cc | 183 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl()
248 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmNnz()
266 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/sparse/ |
| D | mat_mul_op.cc | 834 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
875 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
886 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 1407 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
1418 char transB, const float* alpha, libxsmm_CSR_sparseslice* A_sparse, in wrapper_libxsmm_spmdm_compute_generic_thread()
|