def taken_keeps
bendtorch-core@0.1.0.0 · src/tensor/matmul_PROOF.bend:1316
import bendtorch-core@0.1.0.0/src/tensor/matmul_PROOF.bend as MatmulPROOF
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/elementwise_totality.bend as ElementwiseTotality
import bendtorch-core@0.1.0.0/src/tensor/elementwise_parallel_proof.bend as Parallel
1316def taken_keeps(+shape: Shape.Shape, -a: Core.Tensor<shape>, -kept: Core.Tensor<shape>, -copy: Array<F32>,
1317 -taken: Parallel.Taken<Core.Packed{shape, a}, (Core.Packed{shape, kept}, copy)>, +other: U32) ->
1318 Parallel.Fact<{Core.checked_observe(shape, kept, other) == Core.checked_observe(shape, a, other) : Maybe<&2, F32>}>:
1319 kept_observe(shape, a, kept, {\{Parallel.Taken: depth => balanced => reads => retained => retained} :
1320 Parallel.Taken<Core.Packed{shape, a}, (Core.Packed{shape, kept}, copy)> -> ElementwiseTotality.SameSource(Core.Packed{shape, a}, Core.Packed{shape, kept})}(taken), other)