Skip to main content

interpolate_backward

Function interpolate_backward 

Source
pub fn interpolate_backward<R: CubeRuntime>(
    input: CubeTensor<R>,
    out_grad: CubeTensor<R>,
    _output_size: [usize; 2],
    options: InterpolateOptions,
) -> CubeTensor<R>
Expand description

Backward interpolate operation

Note: only nearest mode is supported