For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

gpu_bicubic_kernel

def gpu_bicubic_kernel[dtype: DType, OutputLayoutType: TensorLayout, output_origin: MutOrigin, InputLayoutType: TensorLayout, input_origin: ImmOrigin](output: TileTensor[dtype, OutputLayoutType, output_origin], input: TileTensor[dtype, InputLayoutType, input_origin])

Perform bicubic interpolation using GPU.

Parameters:

dtype (DType): Element type of the input and output tensors.
OutputLayoutType (TensorLayout): TensorLayout of the output tensor.
output_origin (MutOrigin): Mutable Origin of the output tensor.
InputLayoutType (TensorLayout): TensorLayout of the input tensor.
input_origin (ImmOrigin): Immutable Origin of the input tensor.

Args:

output (TileTensor[dtype, OutputLayoutType, output_origin]): Output tensor with desired dimensions on the device.
input (TileTensor[dtype, InputLayoutType, input_origin]): Input tensor of shape [B, C, H, W] on the device.