Skip to main content

Mojo function

cpu_bicubic_kernel

cpu_bicubic_kernel(output_host: TileTensor[output_host.dtype, output_host.LayoutType, output_host.origin, address_space=output_host.address_space, linear_idx_type=output_host.linear_idx_type, element_size=output_host.element_size], input_host: TileTensor[input_host.dtype, input_host.LayoutType, input_host.origin, address_space=input_host.address_space, linear_idx_type=input_host.linear_idx_type, element_size=input_host.element_size])

Perform bicubic interpolation on a TileTensor of form NCHW.

Args:

  • output_host (TileTensor): Output tensor with desired dimensions.
  • input_host (TileTensor): Input tensor of shape [B, C, H, W].

Was this page helpful?