Skip to main content

Mojo function

resize_bicubic

resize_bicubic[dtype: DType, //, target: StringSlice[StaticConstantOrigin]](output: TileTensor[dtype, output.LayoutType, output.origin, linear_idx_type=output.linear_idx_type, element_size=output.element_size], input: TileTensor[dtype, input.LayoutType, input.origin, linear_idx_type=input.linear_idx_type, element_size=input.element_size], 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?