def Flat.matmul

bend-tensors@0.0.0.2 · bend_tensors.bend:2190

import Base

2190def Flat.matmul(+rows: Nat, +k: Nat, +cols: Nat, a: Flat<rows, k>, b: Flat<k, cols>) -> IO(Flat<rows, cols>):
2191  FlatS{x} = a
2192  FlatS{y} = b
2193  Flat.matmul.run(rows, k, cols, 0, 0, x, y)