Skip to main content

Mojo function

resize_bicubic

resize_bicubic[dtype: DType, rank: Int, output_shape: DimList, input_shape: DimList, //, target: StringSlice[StaticConstantOrigin]](output: NDBuffer[dtype, rank, MutableAnyOrigin, output_shape], input: NDBuffer[dtype, rank, MutableAnyOrigin, input_shape], ctx: DeviceContextPtr)

Perform bicubic interpolation.

Args:

  • output (NDBuffer): Output tensor with desired dimensions on host or device.
  • input (NDBuffer): Input tensor of shape [B, C, H, W] on host or device.
  • ctx (DeviceContextPtr): Device context to enqueue GPU kernels on.

Was this page helpful?