~/jendhub

def step_owner

bendtorch-core@0.1.0.0 · src/autograd/gradient_PROOF.bend:149

import bendtorch-core@0.1.0.0/src/autograd/gradient_PROOF.bend as GradientPROOF
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

149def step_owner(-shape: Shape.Shape, result: Gradient.Step<shape>) -> Core.Dynamic:
150  match result:
151    case Gradient.Stop{contribution, total, error}:
152      contribution
153    case Gradient.Continue{state}:
154      state_contribution(shape, state)

Uses (4)

Used by (8)