Searched refs:n2 (Results 1 – 1 of 1) sorted by relevance
| /libCEED/backends/magma/ |
| H A D | ceed-magma-gemm-nontensor.cpp | 114 magma_int_t n2 = n - (batchCount * nbatch); in magma_gemm_nontensor() local 126 if (n2 > 0) { in magma_gemm_nontensor() 127 …devblas_gemm(trans_A, trans_B, m, n2, k, alpha, d_A, ldda, d_B + batchCount * strideB, lddb, beta,… in magma_gemm_nontensor() 136 if (n2 > 0) { in magma_gemm_nontensor() 137 …devblas_gemm_batched_strided(trans_A, trans_B, m, n2, k, alpha, d_A, ldda, strideA, d_B + batchCou… in magma_gemm_nontensor()
|