def cell_at

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:280

import Base
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel

280# The cell function `Parallel.build` runs: a cell at or past the count is
281# +0.0 padding.
282def cell_at(+layout: Layout, +position: Nat, inputs: Array<F32> & Array<F32>, inside: Bool) -> F32 & (Array<F32> & Array<F32>):
283  (la, ra) = inputs
284  match inside:
285    case False{}:
286      (0.0, (la, ra))
287    case True{}:
288      dot_at(layout, position, la, ra)