~/bend-docscommunity

kernel_gather.bend checks

raw source on the hub · import stelliferous@0.0.2.0/kernel_gather.bend as Kernel_gather

2 imports
import Base
import ./traversal_array.bend as Traversal

Definitions

def gather_width_when source · line 6 · raw

@inside:Bool -> @offset:U32 -> @output_positions:U32 -> @span:U32 -> U32

Clip once per leaf. A channel is one contiguous copy; padding never becomes a store into the next channel. Both owners stay affine across the channel loop.

def gather_width source · line 11 · raw

@+offset:U32 -> @+output_positions:U32 -> @span:U32 -> U32

def gather_channel source · line 14 · raw

@+channel:U32 -> @+width:U32 -> @+offset:U32 -> @+output_positions:U32 -> @+span:U32 -> @owners:Pair(Array<F32>, Array<F32>) -> Pair(Array<F32>, Array<F32>)

def gather_channels source · line 20 · raw

@remaining:Nat -> @+channel:U32 -> @+width:U32 -> @+offset:U32 -> @+output_positions:U32 -> @+span:U32 -> @owners:Pair(Array<F32>, Array<F32>) -> Pair(Array<F32>, Array<F32>)

def gathered_output source · line 28 · raw

@owners:Pair(Array<F32>, Array<F32>) -> Array<F32>

def gather_block source · line 32 · raw

@channels:U32 -> @+offset:U32 -> @+output_positions:U32 -> @+span:U32 -> @input_storage:Array<F32> -> @output_storage:Array<F32> -> Array<F32>