def out_saved

bendtorch@1.3.0.0 · main.bend:1405

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

1405# the output, and a copy of it saved for the backward pass
1406def out_saved(-n: Nat, -c: Nat, r: TA.MM2<n, c>) -> Fwd<Unit, TA.Mat<n, c>, TA.Mat<n, c>>:
1407  match r:
1408    case TA.MM2{a, b}:
1409      Fwd{Unit{}, a, b}