Skip to main content

Mojo function

cpu_bicubic_kernel

cpu_bicubic_kernel[type: DType, rank: Int, //](output_host: NDBuffer[type, rank, origin, shape, strides], input_host: NDBuffer[type, rank, origin, shape, strides])

Perform bicubic interpolation on an NDBuffer of form NCHW.

Args:

  • output_host (NDBuffer[type, rank, origin, shape, strides]): Output tensor with desired dimensions.
  • input_host (NDBuffer[type, rank, origin, shape, strides]): Input tensor of shape [B, C, H, W].

Was this page helpful?