def vjp_targets

bendtorch-core@0.1.0.0 · src/autograd/backward_PROOF.bend:183

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

183def vjp_targets(node: Backward.Node, +g: F32) -> {contribution_targets(Backward.vjp(node, g)) == Laws.targets(node) : List<&2, Nat>}:
184  match node:
185    case Backward.Leaf{}:
186      {==}
187    case Backward.Add{a, b}:
188      {==}
189    case Backward.Sub{a, b}:
190      {==}
191    case Backward.Mul{a, b, left, right}:
192      {==}
193    case Backward.Neg{a}:
194      {==}