def spread

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

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

44def spread(node: Backward.Node, flag: Bool, +count: Nat, flags: List<&2, Bool>) -> List<&2, Bool>:
45  match flag:
46    case False{}:
47      flags
48    case True{}:
49      mark_all(targets(node), count, flags)