def bicubic(x, scale=3): model = Sequential() model.add(InputLayer(input_shape=x.shape[-3:])) model.add(ImageRescale(scale, method=tf.image.ResizeMethod.BICUBIC)) return model