Skip to main content

get_convolution_forward_algorithm

Function get_convolution_forward_algorithm 

Source
pub fn get_convolution_forward_algorithm(
    handle: &Handle,
    src: &TensorDescriptor,
    filter: &FilterDescriptor,
    conv: &ConvolutionDescriptor,
    dst: &TensorDescriptor,
    requested: i32,
) -> Result<Vec<FwdAlgoPerf>>
Expand description

Heuristic-pick the top-N forward-convolution algorithms (cheap; doesn’t run them).