def kept

stelliferous@0.0.2.0 · kernel_gemm_buffer.bend:24

import Base
import stelliferous@0.0.2.0/kernel_gemm.bend as Gemm
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/proofs/kernel_gemm_certificate.bend as GemmCertificate

24def kept(state: Gemm.Buffers,input_length: Nat,weight_length: Nat,bias_length: Nat,length: Nat,certificates: GemmCertificate.Kept<state>) -> Kept:
25  Gemm.Buffers{input,weights,bias,output} = state
26  GemmCertificate.Kept{input_certificate,weight_certificate,bias_certificate,certificate} = certificates
27  Kept{Storage.Buffer{input,input_length,input_certificate},Storage.Buffer{weights,weight_length,weight_certificate},
28    Storage.Buffer{bias,bias_length,bias_certificate},Storage.Buffer{output,length,certificate}}