def read_offset

bendtorch-core@0.1.0.0 · src/tensor/matmul.bend:427

import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matmul

import Base

427def read_offset(+order: Order, +inner: Nat, +outers: Nat, +outer: Nat, +k: Nat) -> Nat:
428  match order:
429    case Kept{}:
430      Nat.add(Nat.mul(outer, inner), k)
431    case Swapped{}:
432      Nat.add(Nat.mul(k, outers), outer)

Uses (4)

Used by (20)