Mojo function
gpu_bicubic_kernel
gpu_bicubic_kernel[dtype: DType, input_layout: Layout, output_layout: Layout, address_space: AddressSpace = AddressSpace(0)](output: LayoutTensor[dtype, output_layout, MutableAnyOrigin, address_space=address_space], input: LayoutTensor[dtype, input_layout, MutableAnyOrigin, address_space=address_space])
Perform bicubic interpolation using GPU.
Args:
- output (
LayoutTensor): Output tensor with desired dimensions on the device. - input (
LayoutTensor): Input tensor of shape [B, C, H, W] on the device.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!