def dense_leaf

stelliferous@0.0.2.0 · tensor_band.bend:357

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage

357def dense_leaf(~Element: Data,geometry: Geometry,fill: Element,+first: U32,+rows: U32,input: Storage.Buffer<Element>) -> Storage.Buffer<Element> & Bands(~Element):
358  Geometry{batch,+channels,+height,+width} = geometry
359  dense_result(~Element,first,rows,
360    planes(~Element,U32.to_nat(channels),0,(rows * width : U32),(height * width : U32),(rows * width : U32),
361      ((first / height * channels * height + first % height) * width : U32),0,(input,allocate(~Element,fill,(channels * rows * width : U32)))))