pub fn bgemm_contiguous_into<T>(
c: &mut ContiguousOperandMut<T>,
a: &ContiguousOperand<T>,
b: &ContiguousOperand<T>,
batch_dims: &[usize],
m: usize,
n: usize,
k: usize,
alpha: T,
beta: T,
) -> Result<()>Expand description
Batched GEMM on pre-contiguous operands.
Operands must already have contiguous inner dimensions (prepared via
prepare_input_* and prepare_output_* in the contiguous module).
batch_dims: sizes of the batch dimensionsm: fused lo dimension size (number of rows of A/C)n: fused ro dimension size (number of cols of B/C)k: fused sum dimension size (inner dimension)