def batched_second

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

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/batched_matmul.bend as Batched

595def batched_second(+b: Nat, +right: Shape.Shape, +plan: Batched.Plan, x: Core.Dynamic, g: Core.Dynamic,
596  pair: Core.Dynamic & Result<&2, &1, Error, Contribution>) -> Pass:
597  (y, first) = pair
598  batched_third(b, right, plan, first, y, x, Core.clone_dynamic(g))