Home
last modified time | relevance | path

Searched refs:transB (Results 1 – 13 of 13) sorted by relevance

/external/armnn/src/armnnOnnxParser/test/
DGemm.cpp18 const std::string& transB, in GemmFixture()
50 name: "transB" in GemmFixture()
51 i: )" + transB + R"( in GemmFixture()
233 name: "transB" in GemmConstantFixture()
352 name: "transB" in GemmConstantSimpleFixture()
446 const std::string& transB, in GemmABFixture()
476 name: "transB" in GemmABFixture()
477 i: )" + transB + R"( in GemmABFixture()
/external/tensorflow/tensorflow/core/util/
Drocm_sparse.cc191 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl() argument
197 hipsparse_handle, transA, transB, m, n, k, nnz, AsHipComplex(alpha_host), in CsrmmImpl()
206 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, \
214 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
226 hipsparseOperation_t transA, hipsparseOperation_t transB, \
233 *gpusparse_handle_, transA, transB, alpha, matA, matB, beta, matC, \
243 hipsparseOperation_t transA, hipsparseOperation_t transB, \
250 wrap::hipsparseSpMM(*gpusparse_handle_, transA, transB, alpha, matA, \
295 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmNnz() argument
304 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
[all …]
Dcuda_sparse.cc394 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, in CsrmmImpl() argument
407 cusparse_handle, transA, transB, m, n, k, nnz, AsCudaComplex(alpha_host), in CsrmmImpl()
416 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, \
424 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
436 cusparseOperation_t transA, cusparseOperation_t transB, \
443 *gpusparse_handle_, transA, transB, alpha, matA, matB, beta, matC, \
453 cusparseOperation_t transA, cusparseOperation_t transB, \
459 transB, alpha, matA, matB, beta, \
764 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmNnz() argument
773 *gpusparse_handle_, transA, transB, m, k, n, descrA, nnzA, in CsrgemmNnz()
[all …]
Dcuda_sparse.h282 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
295 gpusparseOperation_t transB, const Scalar* alpha,
308 Status SpMM(gpusparseOperation_t transA, gpusparseOperation_t transB,
393 Status CsrgemmNnz(gpusparseOperation_t transA, gpusparseOperation_t transB,
419 Status Csrgemm(gpusparseOperation_t transA, gpusparseOperation_t transB,
/external/eigen/bench/
Dbench_gemm.cpp81 const char transB = trans; variable
83 const char transB = notrans; variable
92 sgemm_(&transA,&transB,&M,&N,&K,&fone, in blas_gemm()
104 dgemm_(&transA,&transB,&M,&N,&K,&done, in blas_gemm()
116 cgemm_(&transA,&transB,&M,&N,&K,(float*)&cfone, in blas_gemm()
128 zgemm_(&transA,&transB,&M,&N,&K,(double*)&cdone, in blas_gemm()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc841 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
861 transA, transB, &alpha, matA, matB, &beta, matC, in Compute()
869 TF_RETURN_IF_ERROR(cuda_sparse.SpMM(transA, transB, &alpha, matA, matB, in Compute()
879 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
899 transA, transB, &alpha, matA, matB, &beta, matC, in Compute()
907 TF_RETURN_IF_ERROR(cuda_sparse.SpMM(transA, transB, &alpha, matA, matB, in Compute()
919 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
930 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
941 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcusparse_10_0.inc2018 cusparseOperation_t transB, int m, int n, int k, int nnz,
2029 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2036 cusparseOperation_t transB, int m, int n, int k, int nnz,
2048 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2055 cusparseOperation_t transB, int m, int n, int k, int nnz,
2067 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2074 cusparseOperation_t transB, int m, int n, int k, int nnz,
2087 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2094 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2106 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_10_1.inc2028 cusparseOperation_t transB, int m, int n, int k, int nnz,
2039 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2046 cusparseOperation_t transB, int m, int n, int k, int nnz,
2058 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2065 cusparseOperation_t transB, int m, int n, int k, int nnz,
2077 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2084 cusparseOperation_t transB, int m, int n, int k, int nnz,
2097 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2104 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2116 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_10_2.inc2028 cusparseOperation_t transB, int m, int n, int k, int nnz,
2039 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2046 cusparseOperation_t transB, int m, int n, int k, int nnz,
2058 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2065 cusparseOperation_t transB, int m, int n, int k, int nnz,
2077 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2084 cusparseOperation_t transB, int m, int n, int k, int nnz,
2097 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2104 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2116 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_11_0.inc1440 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1452 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1459 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1471 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1478 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1491 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1498 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1512 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1604 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
1616 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcusparse_9_0.inc2018 cusparseOperation_t transB, int m, int n, int k, int nnz,
2029 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2036 cusparseOperation_t transB, int m, int n, int k, int nnz,
2048 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2055 cusparseOperation_t transB, int m, int n, int k, int nnz,
2067 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2074 cusparseOperation_t transB, int m, int n, int k, int nnz,
2087 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2094 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2106 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1416 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread() argument
1420 handle, transA, transB, reinterpret_cast<const libxsmm_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() argument
1430 return libxsmm_spmdm_compute_fp32_thread(handle, transA, transB, alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1994 int transB = static_cast<int>(ReadOptionalNodeUint32Attribute(node, "transB", 0)); in ParseGemm() local
2005 fullyConnectedDescriptor.m_TransposeWeightMatrix = transB; in ParseGemm()