Skip to main content

matmul_spec

Function matmul_spec 

Source
pub fn matmul_spec(
    lhs: Operand,
    rhs: Operand,
    output: Operand,
    m: u32,
    n: u32,
    k: u32,
    batch: u32,
) -> Vec<u32>
Expand description

Specialization payload of a MATMUL: lhs, rhs, output, then m, n, k, batch.