def cell_of

bendtorch-core@0.1.0.0 · src/tensor/matmul_parallel_proof.bend:180

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix
import bendtorch-core@0.1.0.0/src/tensor/elementwise_parallel_proof.bend as Parallel

180# The matmul cell function of `Parallel.build`, from the input arrays.
181def cell_of(+inner: Nat, +columns: Nat, +position: Nat, A: Array<F32>, B: Array<F32>, inside: Bool) -> F32 & (Array<F32> & Array<F32>):
182  Matrix.cell_at(Matrix.dims(Matrix.Dims{inner, columns}), position, (A, B), inside)