bgemm_strided_into_with_map

Function bgemm_strided_into_with_map 

Source
pub fn bgemm_strided_into_with_map<T, MapA, MapB>(
    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,
    map_a: MapA,
    map_b: MapB,
) -> Result<()>
where T: Copy + Mul<Output = T> + Add<Output = T> + Zero + One + PartialEq, MapA: Fn(T) -> T, MapB: Fn(T) -> T,
Expand description

Batched strided GEMM with closure-based element mapping: C = alpha * map_a(A) * map_b(B) + beta * C

Like bgemm_strided_into but uses closures instead of conjugation flags, allowing custom scalar types that don’t implement ElementOpApply.