tfm.vision.spatial_transform_ops.bilinear_resize

Bilinear resizes the images.

images A tensor in shape (batch_size, input_h, input_w, ...) with arbitrary numbers of channel dimensions.
output_size The size of the output image in (output_h, output_w).

A tensor in shape (batch_size, output_h, output_w, ...). The result has the same dtype as the input if it's float32, float16, bfloat16, otherwise the result is float32.