def slots_of

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward_LAWS.bend:22

import Base
import bendtorch-core@0.1.0.0/src/autograd/gradient.bend as Gradient
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as Backward

22def slots_of(entries: List<&1, Backward.Entry>) -> List<&1, Gradient.Slot>:
23  match entries:
24    case Nil{}:
25      Nil{}
26    case Con{Backward.Entry{node, shape, value, saved, slot}, rest}:
27      slot <> slots_of(rest)