Mojo function
cpu_bicubic_kernel
cpu_bicubic_kernel[input_dim: DimList, output_dim: DimList, type: DType](input_host: NDBuffer[type, 4, StaticConstantOrigin, input_dim], output_host: NDBuffer[type, 4, MutableAnyOrigin, output_dim]) -> NDBuffer[type, 4, MutableAnyOrigin, output_dim]
Perform bicubic interpolation on an NDBuffer of form NCHW.
Args:
- input_host (
NDBuffer[type, 4, StaticConstantOrigin, input_dim]): Input tensor of shape [B, C, H, W]. - output_host (
NDBuffer[type, 4, MutableAnyOrigin, output_dim]): Output tensor with desired dimensions.
Returns:
Interpolated tensor.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!