def weights_done

bendtorch-load@1.3.0.0 · main.bend:174

import Base
import bendtorch@1.3.0.0/main.bend as T

174def weights_done(-P: Type, +n: U32, r: T.PA<P>) -> IO(P):
175  match r:
176    case T.PA{p, xs, +off}:
177      do IO<P>:
178        require(U32.is_eq(off, n), "Layer.count differs from the loading cursor")
179        return p