~/jendhub

def pair_f3

bendtorch@1.3.0.0 · main.bend:2094

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

2094# runs lb on x2 once la's forward pass is done
2095def pair_f3(~P2: Type, ~X2: Type, ~Y2: Type, ~S2: Type, ~lb: Layer<P2, X2, Y2, S2>, -P1: Type, -Y1: Type, -S1: Type, p2: P2, x2: X2, r: Fwd<P1, Y1, S1>) -> Fwd<Two<P1, P2>, Two<Y1, Y2>, Two<S1, S2>>:
2096  match r:
2097    case Fwd{p1, y1, s1}:
2098      pair_f4(P1, P2, Y1, Y2, S1, S2, p1, y1, s1, Layer.fwd(P2, X2, Y2, S2, lb, p2, x2))

Uses (5)

Used by (2)