def leaf

stelliferous@0.0.2.0 · kernel_band_pipeline.bend:32

import Base
import stelliferous@0.0.2.0/tensor_band.bend as Bands
import stelliferous@0.0.2.0/tensor_band_window.bend as BandWindow
import stelliferous@0.0.2.0/kernel_tile.bend as Tile
import stelliferous@0.0.2.0/storage_buffer.bend as Storage

32def leaf(~custom: F32 -> F32,context: Context,+first: U32,+rows: U32,core: Storage.Buffer<F32>,halo: BandWindow.Halo<F32>,
33  weights: Storage.Buffer<Tile.Tile8>,bias: Storage.Buffer<F32>) -> Bands.Bands(~F32):
34  Context{+geometry,+channels,+inner,parallel,activation} = context
35  BandWindow.Halo{above,below,+rows_read} = halo
36  windowed(~custom,activation,geometry,channels,inner,first,rows,rows_read,core,above,below,weights,bias)