pub fn bgemm_strided_into<T>(
c: &mut StridedViewMut<'_, T>,
a: &StridedView<'_, T>,
b: &StridedView<'_, T>,
_n_batch: usize,
n_lo: usize,
n_ro: usize,
n_sum: usize,
alpha: T,
beta: T,
conj_a: bool,
conj_b: bool,
) -> Result<()>Expand description
Batched strided GEMM: C = alpha * A * B + beta * C
The views must be pre-permuted so that their dimensions are grouped as (batch-last canonical order):
- A:
n_lodims, thenn_sumdims, thenn_batchbatch dims - B:
n_sumdims, thenn_rodims, thenn_batchbatch dims - C:
n_lodims, thenn_rodims, thenn_batchbatch dims
Dimension sizes must match across operands within each group.