def band_work

stelliferous@0.0.2.0 · tensor_convolution.bend:139

import Base
import stelliferous@0.0.2.0/kernel_planning.bend as KernelPlanning
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_band_pipeline.bend as BandPipeline

139def band_work(context: BandPipeline.Context,bias_capacity: U32) -> KernelPlanning.Work:
140  BandPipeline.Context{BandWindow.Geometry{Bands.Geometry{batch,+channels,+height,+width},kernel,stride,padding,oh,+ow},rows,inner,parallel,activation} = context
141  KernelPlanning.Work{(oh * ow : U32),inner,rows,KernelPlanning.Owners{U32.from_nat(KernelPlanning.capacity((channels * height * width : U32))),bias_capacity},
142    KernelPlanning.Geometry{channels,height,width,kernel,stride,padding,ow},activation}