def CellOut

bendtorch-core@0.1.0.0 · src/tensor/elementwise_parallel_proof.bend:110

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise

110# What an inside cell of the build gives: both arrays back, and the F32
111# `want` names.
112def CellOut(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, +dl: Nat, +dr: Nat, -A: Array<F32>, -B: Array<F32>, +position: Nat,
113  -result: F32 & (Array<F32> & Array<F32>)) -> Type:
114  {Pair.snd(F32, Array<F32> & Array<F32>, result) == (A, B) : Array<F32> & Array<F32>} &
115    {Some{Pair.fst(F32, Array<F32> & Array<F32>, result)} == want(operation, plan, dl, dr, A, B, position) : Maybe<&2, F32>}