def pair_finish

stelliferous@0.0.2.0 · tensor_lanes.bend:57

import stelliferous@0.0.2.0/tensor_lanes.bend as TensorLanes
import Base
import stelliferous@0.0.2.0/traversal_cells.bend as Cells
import stelliferous@0.0.2.0/kernel_tile.bend as Tile
import stelliferous@0.0.2.0/tensor_operations.bend as Operations
import stelliferous@0.0.2.0/tensor_view.bend as Views

57# The checked caller has already validated both views and owners.
58def pair_finish(~operation: F32 -> F32 -> F32,count: Nat,state: Cells.State<Binary,Tile.Tile8,Unit>) -> Binary:
59  Cells.State{Binary{left,right},scratch,+position,slot,context} = state
60  +count = count
61  pair_result(Operations.combine_kernel(~F32,~operation,count,Views.View{[Views.Axis{U32.from_nat(count),1}],position},position,left,right))

Uses (12)

Used by (1)