def program_fan_out

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

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

175# x = 0, a = 1, b = 2, c = 3, p = x * a = 4, q = x * b = 5, r = x * c = 6,
176# s = p + q = 7, t = s + r = 8. x has three consumers.
177def program_fan_out(x: F32, a: F32, b: F32, c: F32) -> Backward.Tape:
178  recorded(Backward.add(recorded(Backward.add(recorded(Backward.mul(recorded(Backward.mul(recorded(Backward.mul(
179    tape_of(Backward.leaf(tape_of(Backward.leaf(tape_of(Backward.leaf(tape_of(Backward.leaf(Backward.empty(), x)), a)), b)), c)),
180    0n, 1n)), 0n, 2n)), 0n, 3n)), 4n, 5n)), 7n, 6n))

Uses (8)

Used by (1)