Skip to main content

Mojo function

resize_bicubic

resize_bicubic[dtype: DType, //, target: StringSlice[StaticConstantOrigin]](output: TileTensor[dtype, LayoutType, origin, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], input: TileTensor[dtype, LayoutType, origin, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], ctx: DeviceContextPtr)

Perform bicubic interpolation.

Args:

  • โ€‹output (TileTensor): Output tensor with desired dimensions on host or device.
  • โ€‹input (TileTensor): 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?