Searched defs:transA (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()
216 hipsparseOperation_t transA, int m, int n, in CsrmvImpl()
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()
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/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 824 const gpusparseOperation_t transA = CUSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
826 const gpusparseOperation_t transA = HIPSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 1389 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1397 char transA, const bfloat16* A, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1407 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
1417 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_compute_generic_thread()
|