def run

stelliferous@0.0.2.0 · kernel_pipeline.bend:205

import Base
import stelliferous@0.0.2.0/kernel_gemm_buffer.bend as GemmBuffer
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/math_activation.bend as Activations

205def run(~custom: F32 -> F32,activation: Activations.Activation,depth: U32,+work: Work,input: Storage.Buffer<F32>,weights: Storage.Buffer<F32>,bias: Storage.Buffer<F32>) -> Storage.Buffer<F32>:
206  Work{rows,h,w,k,s,p,ow,positions,inner} = work
207  run_packed(~custom,activation,depth,work,input,GemmBuffer.reorder(rows,inner,weights),bias)