def gather_sent

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:658

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/tensor/gather.bend as Gather

658def gather_sent(+a: Nat, +input: Shape.Shape, pair: Core.Dynamic & Result<&2, &1, Gather.Error, Core.Dynamic>) -> Pass:
659  (g, result) = pair
660  match result:
661    case Fail{error}:
662      Pass{Unsaved{}, g, Fail{GatherError{error}}}
663    case Done{scattered}:
664      Pass{Unsaved{}, g, Done{[Contribution{a, input, scattered}]}}