def planned_model

bendtorch-core@0.1.0.0 · src/tensor/transform_PROOF.bend:1045

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms

1045def planned_model(+input: Shape.Shape, +axes: List<&2, U32>, source: Core.Tensor<input>,
1046  planned: Result<&2, &2, Layout.Error, Layout.Plan<input, axes>>) ->
1047  Transforms.PublicModel<input, axes, source, Transforms.copy_planned(input, axes, source, planned)>:
1048  match planned:
1049    case Fail{error}:
1050      Transforms.PublicModel{Unit{}}
1051    case Done{+plan}:
1052      allocated_model(input, axes, plan, source, Core.new(Layout.output(input, axes, plan), 0.0))