kernel_packing_samples.bend checks

raw source on the hub · import stelliferous@0.0.2.0/kernel_packing_samples.bend as Kernel_packing_samples

Pixel reads of an entry whose tile crosses an output row.

2 imports

import Base
import ./kernel_tile.bend as Tile

Types

type PixelBlock source · line 22 · raw

Data

PixelBlock

@position:U32 -> @kernel_row:U32 -> @kernel_column:U32 -> @channel_base:U32 -> @height:U32 -> @width:U32 -> @stride:U32 -> @padding:U32 -> @output_width:U32 -> PixelBlock

type PixelAddress source · line 31 · raw

Data

PixelAddress

@index:U32 -> @inside:Bool -> PixelAddress

Definitions

def masked source · line 5 · raw

@value:F32 -> @inside:Bool -> F32

def mask_pixel source · line 12 · raw

@inside:Bool -> @pair:Pair(Array<F32>, F32) -> Pair(Array<F32>, F32)

def pixel source · line 16 · raw

@input_storage:Array<F32> -> @+output_position:U32 -> @+reduction_index:U32 -> @+input_height:U32 -> @+input_width:U32 -> @+kernel_size:U32 -> @+stride:U32 -> @+padding:U32 -> @+output_width:U32 -> Pair(Array<F32>, F32)

def pixel_block source · line 27 · raw

@position:U32 -> @+reduction:U32 -> @+height:U32 -> @+width:U32 -> @+kernel:U32 -> @stride:U32 -> @padding:U32 -> @output_width:U32 -> PixelBlock

def pixel_address source · line 34 · raw

@context:PixelBlock -> @index:Nat -> PixelAddress

def read_pixel_address source · line 41 · raw

@input:Array<F32> -> @address:PixelAddress -> Pair(Array<F32>, F32)

def read_block_pixel source · line 45 · raw

@input:Array<F32> -> @context:PixelBlock -> @index:Nat -> Pair(Array<F32>, F32)