Skip to main content

pool

Function pool 

Source
pub fn pool(
    kind: PoolKind,
    input_shape: &[i64],
    kernel_shape: &[i64],
    padding: &[i64],
    stride_shape: &[i64],
    output_shape: &[i64],
    input: &[f32],
    output: &mut [f32],
)
Expand description

Execute an N-dimensional NCHW Float32 pool using MLAS.