def tie_bwd

bendtorch@1.3.0.0 · main.bend:2199

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

2199def tie_bwd(~P: Type, ~Q: Type, ~X: Type, ~Y: Type, ~Z: Type, ~W: Type, ~S1: Type, ~S2: Type, ~S3: Type, ~la: Layer<P, X, Y, S1>, ~lb: Layer<Q, Y, Z, S2>, ~lc: Layer<P, Z, W, S3>, pq: Two<P, Q>, s: Three<S1, S2, S3>, dw: W, g: Two<P, Q>) -> Bwd<Two<P, Q>, X>:
2200  match pq s g:
2201    case Two{p, q} Three{s1, s2, s3} Two{gp, gq}:
2202      tie_b2(~P, ~Q, ~X, ~Y, ~Z, ~S1, ~S2, ~la, ~lb, q, s1, s2, gq, Layer.bwd(P, Z, W, S3, lc, p, s3, dw, gp))

Uses (6)

Used by (1)