def targets

bendtorch-core@0.1.0.0 · src/autograd/backward_LAWS.bend:24

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

24def targets(node: Backward.Node) -> List<&2, Nat>:
25  match node:
26    case Backward.Leaf{}:
27      []
28    case Backward.Add{a, b}:
29      [a, b]
30    case Backward.Sub{a, b}:
31      [a, b]
32    case Backward.Mul{a, b, left, right}:
33      [a, b]
34    case Backward.Neg{a}:
35      [a]