def Final

bendtorch-core@0.1.0.0 · src/tensor/permute_values.bend:53

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

53def Final(+input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +fill: F32, +ops: List<&2, StorageLaws.Op>, +count: Nat,
54  -Q: Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>> -> Type) -> Type:
55  @-source_fits: {Lifecycle.logical_length(kept_source(input, fill, ops)) == Shape.logical_numel(Shape.dimensions(input)) : Nat} ->
56  @-target_fits: {Lifecycle.logical_length(written(input, axes, plan, fill, ops, count)) ==
57    Shape.logical_numel(Shape.dimensions(Layout.output(input, axes, plan))) : Nat} ->
58    Q((Core.TTensor{kept_source(input, fill, ops), source_fits}, Done{Core.TTensor{written(input, axes, plan, fill, ops, count), target_fits}}))