def nodes_of

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

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

14# The nodes of a list of entries, newest first.
15def nodes_of(entries: List<&1, Backward.Entry>) -> List<&2, Backward.Node>:
16  match entries:
17    case Nil{}:
18      Nil{}
19    case Con{Backward.Entry{node, shape, value, saved, slot}, rest}:
20      node <> nodes_of(rest)