在哪里可以确定哪些Keras方法是TPU兼容方法？

Question

我最近了解到UpSampling2D不是TPU支持的操作。

我如何确定哪些Keras功能与Colab中的TPU兼容？我找不到有关此主题的任何文档。

Answer 1

通常应该支持所有Keras层，但是没有一个很好的摘要。有一个与TPU Ops here相关的更一般的摘要。

尤其是UpSampling2D目前尚在开发中，但是如果您在Keras和TPU上遇到更多问题，应鼓励您提出问题here。