def seeded_run

bendtorch-core@0.1.0.0 · src/autograd/tensor_chain_PROOF.bend:544

import bendtorch-core@0.1.0.0/src/autograd/tensor_chain_PROOF.bend as TensorChainPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/autograd/gradient.bend as Gradient
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as Backward

544def seeded_run(+node: Backward.Node, +shape: Shape.Shape, value: Core.Dynamic, saved: Backward.Saved, slot: Gradient.Slot,
545  rest: List<&1, Backward.Entry>, +older: Nat, -ones: Core.Tensor<shape>,
546  made: {Core.new(shape, 1.0) == Done{ones} : Result<&2, &1, Core.Error, Core.Tensor<shape>>},
547  init: Started<Gradient.Held{Core.Packed{shape, ones}} <> Backward.unsets(older), 1n+older>) ->
548  Passed<Backward.Entry{node, shape, value, saved, slot} <> rest, 1n+older, older, True{},
549    Gradient.Held{Core.Packed{shape, ones}} <> Backward.unsets(older)>:
550  Started{copy, same, pulls} = init
551  seeded_out(node, shape, value, saved, slot, rest, older, ones, copy, made, same,
552    swept_pulls(Backward.Entry{node, shape, value, saved, slot} <> rest, Done{copy},
553      Gradient.Held{Core.Packed{shape, ones}} <> Backward.unsets(older), 1n+older, Nil{}, pulls))

Uses (18)

Used by (1)