~/jendhub

def ts2

bendtorch@1.3.0.0 · main.bend:941

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

941def ts2(~P: Type, ~X: Type, ~Y: Type, ~S: Type, ~T: Type, ~m: Layer<P, X, Y, S>, ~loss: Y -> T -> LossOut<Y>, +cfg: OptCfg, g: P, st: OptState<P>, +p1: F32, +p2: F32, t: T, r: Fwd<P, Y, S>) -> Train<P>:
942  match r:
943    case Fwd{p, y, s}:
944      ts3(~P, ~X, ~Y, ~S, ~m, cfg, p, s, g, st, p1, p2, loss(y, t))

Uses (8)

Used by (1)