def block_or_tree
bendtorch-core@0.1.0.0 · src/tensor/matmul.bend:590
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
590def block_or_tree(+layout: Layout, +count: Nat, +base: Nat, state: Array<F32> & Array<F32>, one: Bool) ->
591 Array<F32> & (Array<F32> & Array<F32>):
592 match one:
593 case False{}:
594 Parallel.tree(~Layout, ~Array<F32>, ~Array<F32>, ~product_cell_at, layout, 3n, count, base, state)
595 case True{}:
596 row_block(layout, base, state)