Skip to main content

separable_conv2d_nhwc

yscv_kernels

Function separable_conv2d_nhwc 

Source 
pub fn separable_conv2d_nhwc(
    input: &Tensor,
    params: SeparableConv2dParams<'_>,
    stride_h: usize,
    stride_w: usize,
) -> Result<Tensor, KernelError>
Expand description

NHWC separable convolution without padding: depthwise ([KH, KW, C, depth_multiplier]) then pointwise ([1, 1, C*depth_multiplier, C_out]).