~/jendhub

def cell_at

bendtorch-core@0.1.0.0 · src/tensor/matmul.bend:299

import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matmul
import Base
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel

299# The cell function `Parallel.build` runs: its state is the pair of input
300# arrays, split for a fork as `Elementwise.split` splits them.
301def cell_at(+dims: Dims, +position: Nat, inputs: Array<F32> & Array<F32>, inside: Bool) -> F32 & (Array<F32> & Array<F32>):
302  Dims{+inner, +columns} = dims
303  (la, ra) = inputs
304  cell(inner, columns, position, la, ra, inside)

Uses (8)

Used by (9)