def read_offset_spec

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_PROOF.bend:370

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul_LAWS.bend as Laws

370def read_offset_spec(+order: Matrix.Order, +inner: Nat, +outers: Nat, +outer: Nat, +k: Nat) ->
371  {Matrix.read_offset(order, inner, outers, outer, k) == Laws.matrix_offset(order, inner, outers, outer, k) : Nat}:
372  match order:
373    case Matrix.Kept{}:
374      {==}
375    case Matrix.Swapped{}:
376      {==}