def dims_planned

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

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/sum_dims.bend as SumDims

1389def dims_planned(+a: Nat, entries: List<&1, Entry>, input: Core.Dynamic, +shape: Shape.Shape, keepdim: Bool,
1390  planned: Result<&2, &2, SumDims.Error, SumDims.Plan>) -> Tape & Result<&2, &2, Error, Nat>:
1391  match planned:
1392    case Fail{error}:
1393      (Tape{entries}, Fail{dims_error(error)})
1394    case Done{SumDims.Plan{+kept, +dropped}}:
1395      dims_summed(SumDims{a, shape, kept}, entries, SumDims.sum_planned(input, SumDims.Plan{kept, dropped}, keepdim))