Skip to main content

Mojo function

cpu_bicubic_kernel

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

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?