Expand description
GEMM, batched GEMM, ragged batched GEMM, and matvec variants.
Functionsยง
- bmm_f32
- bmm_f64
- group_
gemm_ f32 - grouped_
gemm_ f32 - Grouped expert GEMM with optional input/output permutation metadata.
- masked_
bmm_ f32 - matmul_
f32 - matmul_
f64 - ragged_
bmm_ f32