bgemm_contiguous_into

Function bgemm_contiguous_into 

Source
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<()>
where T: ComplexField + Copy + ElementOpApply + Send + Sync + Mul<Output = T> + Add<Output = T> + Zero + One + PartialEq,
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 dimensions
  • m: 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)