def wake_cost

stelliferous@0.0.2.0 · kernel_band_planning.bend:87

import Base
import stelliferous@0.0.2.0/kernel_cost.bend as Cost
import stelliferous@0.0.2.0/tensor_band.bend as Bands
import stelliferous@0.0.2.0/tensor_band_window.bend as BandWindow

87def wake_cost(window: BandWindow.Geometry,config: Cost.Configuration) -> F32:
88  BandWindow.Geometry{Bands.Geometry{batch,channels,height,width},kernel,stride,padding,oh,ow} = window
89  Cost.Configuration{mac,byte_ns,operation_ns,grain,wake} = config
90  (2.0 * U32.to_f32(batch) * wake : F32)