def eight_leaves_of

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

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

558# The eight words as the depth-3 tree `Parallel.tree` builds.
559def eight_leaves_of(+cells: Eight, state: Array<F32> & Array<F32>) -> Array<F32> & (Array<F32> & Array<F32>):
560  Eight{+w0, +w1, +w2, +w3, +w4, +w5, +w6, +w7} = cells
561  (ANode{ANode{ANode{ALeaf{w0}, ALeaf{w1}}, ANode{ALeaf{w2}, ALeaf{w3}}}, ANode{ANode{ALeaf{w4}, ALeaf{w5}}, ANode{ALeaf{w6}, ALeaf{w7}}}}, state)

Uses (5)

Used by (3)