是否有任何解决方案upsampling2d（与tpu协作）

Question

第一次问问题：
谷歌合作实验室是伟大的和速度，但我发现何时使用TPU。
它不支持升采样方法，
像：tf.keras.layers.Upsampling2D或tf.image.resize_images
我也尝试使用lambda，但它也不起作用

只想知道是否有解决方案或替代方法？ （我不知道为什么colab尚未实现此基本层）

感谢您的收听，欢迎您提供所有反馈意见：）

    ...
    for i in range(2):
        x=Conv2D(256,kernel_size=3,strides=(1,1),padding='same')(x)
    >>  x=Lambda(upsampling2d_tpu)(x)    
        x=Activation('relu')(x)
    x_out=Conv2D(3,kernel_size=9,strides=(1,1),padding='same')(x)
    model=Model(x_in,x_out)
    return model

def upsampling2d_tpu(inputs):
    _,h,w,_=inputs.shape
    x=tf.image.resize_images(inputs,(2*h,2*w))
    return x

错误消息： 编译失败：编译失败：尝试在XLA_TPU_JIT上编译图cluster_2_11289656269171445970 []时检测到不受支持的操作：ResizeNearestNeighbor（未注册XLA_TPU_JIT的'ResizeNearestNeighbor'OpKernel ...