Bendtorch.Train.dump: def of bendtorch@1.3.0.0
Import: import bendtorch@1.3.0.0/main.bend as Bendtorch
Lines 3504-3515 of bendtorch@1.3.0.0/main.bend: https://jend.leeeet.dev/read.txt?path=bendtorch@1.3.0.0/main.bend&start_line=3504&end_line=3515

import Base

# every number of a training state: the parameters, then the first moments, the second moments, b1^t and b2^t (packed
# moments: their tree's cells, pt_store's layout, in place of the two moments)
def Train.dump(~P: Type, ~X: Type, ~Y: Type, ~S: Type, ~m: Layer<P, X, Y, S>, tr: Train<P>) -> TrainPL<P>:
  match tr:
    case Train{p, g, NoMom{}, +p1, +p2, +l}:
      td0(P, g, p1, p2, l, Layer.dump(P, X, Y, S, m, p))
    case Train{p, g, OneMom{mm}, +p1, +p2, +l}:
      td1(P, g, p1, p2, l, Layer.dump(P, X, Y, S, m, p), Layer.dump(P, X, Y, S, m, mm))
    case Train{p, g, TwoMom{mm, vv}, +p1, +p2, +l}:
      td3(P, g, p1, p2, l, Layer.dump(P, X, Y, S, m, p), Layer.dump(P, X, Y, S, m, mm), Layer.dump(P, X, Y, S, m, vv))
    case Train{p, g, Packed{t}, +p1, +p2, +l}:
      tdp(P, g, p1, p2, l, Layer.dump(P, X, Y, S, m, p), pt_cells(t))

Uses (10):
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=Layer
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=Layer.dump
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=OptState
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=Train
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=TrainPL
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=pt_cells
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=td0
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=td1
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=td3
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=tdp