Skip to main content

conv1d_gemm

Function conv1d_gemm 

Source
pub fn conv1d_gemm(
    x: &[f32],
    w: &[f32],
    ic: usize,
    oc: usize,
    n: usize,
    k: usize,
    pad: usize,
    dil: usize,
    out_n: usize,
    yt: &mut [f32],
) -> bool
Expand description

A 1D convolution as a GEMM whose column matrix is expanded on the device instead of being built, transposed and uploaded by the host. yt comes back [out_n x oc]. false = refused, caller runs host.