bgemm_strided_into

Function bgemm_strided_into 

Source
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<()>
where T: Copy + ElementOpApply + Mul<Output = T> + Add<Output = T> + Zero + One + PartialEq,
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_lo dims, then n_sum dims, then n_batch batch dims
  • B: n_sum dims, then n_ro dims, then n_batch batch dims
  • C: n_lo dims, then n_ro dims, then n_batch batch dims

Dimension sizes must match across operands within each group.