协作中TPU的UpSampling2D层版本

Question

看来Keras中的UpSampling2D层与Google Collab中的TPU不兼容。是否有可能创建与TPU兼容且执行相同操作的自定义层？