def execute_many

stelliferous@0.0.2.0 · tensor_convolution.bend:168

import stelliferous@0.0.2.0/tensor_convolution.bend as TensorConvolution
import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/kernel_tile.bend as Tile
import stelliferous@0.0.2.0/default_kernel_cost.bend as DefaultCost
import stelliferous@0.0.2.0/proofs/storage_certificate.bend as Certificate
import stelliferous@0.0.2.0/kernel_band_planning.bend as BandPlanning
import stelliferous@0.0.2.0/math_activation.bend as Activations

168def execute_many(~custom: F32 -> F32,+activation: Activations.Activation,+geometry: Geometry,input: Storage.Buffer<F32>,weights: Storage.Buffer<Tile.Tile8>,bias: Storage.Buffer<F32>,+workers: U32) -> Tensors.Tensor<F32>:
169  Storage.Buffer{bias,length,Certificate.Certificate{+depth,equation}} = bias
170  +context = band_context(geometry,True{},activation)
171  initial_layout(~custom,activation,BandPlanning.select(band_work(context,U32.shln(1,depth)),context_window(context),workers,DefaultCost.configuration()),
172    geometry,input,weights,Storage.Buffer{bias,length,Certificate.Certificate{depth,equation}},workers)

Uses (16)

Used by (1)