Skip to main content

Mojo function

resize_bicubic

resize_bicubic[type: DType, rank: Int, //, target: StringSlice[StaticConstantOrigin]](output: NDBuffer[type, rank, origin, shape, strides], input: NDBuffer[type, rank, origin, shape, strides], ctx: DeviceContextPtr)

Perform bicubic interpolation.

Args:

  • output (NDBuffer[type, rank, origin, shape, strides]): Output tensor with desired dimensions on host or device.
  • input (NDBuffer[type, rank, origin, shape, strides]): 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?