def reordered

stelliferous@0.0.2.0 · tensor_convolution.bend:194

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/kernel_shape.bend as KernelShape
import stelliferous@0.0.2.0/kernel_gemm_buffer.bend as GemmBuffer
import stelliferous@0.0.2.0/kernel_tile.bend as Tile

194# The weights in the order the kernel reads them: once, after the checks.
195def reordered(dimensions: KernelShape.Dimensions,weights: Storage.Buffer<F32>) -> Storage.Buffer<Tile.Tile8>:
196  KernelShape.Dimensions{ci,rows,h,w,+k,s,p} = dimensions
197  GemmBuffer.reorder(U32.from_nat(rows),U32.from_nat(Nat.mul(ci,Nat.mul(k,k))),weights)