def prepared

stelliferous@0.0.2.0 · tensor_band_pool.bend:76

import stelliferous@0.0.2.0/tensor_band_pool.bend as TensorBandPool
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

76def prepared(~Element: Data,~operation: Element -> Element -> Element,+context: Context<Element>,+layout: Bands.Plan,owners: Bands.Bands(~Element) & BandWindow.Halos(~Element)) -> Bands.Bands(~Element):
77  Context{+window,pad,fill,workers} = context
78  (bands,halos) = owners
79  +geometry = BandWindow.input_geometry(window)
80  +enabled = parallel(window,workers,plan_size(layout,window))
81  BandWindow.apply(~Element,~Context<Element>,~leaf(~Element,~operation),layout,enabled && Bands.same_image(geometry,layout),enabled,geometry,context,bands,halos)

Uses (13)

Used by (1)