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

Question

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

如何找出哪些 Keras 函数与 Colab 中的 TPU 兼容？我无法找到有关该主题的任何文档。

Answer 1

一般来说所有的 Keras 层都应该支持，但是没有一个很好的总结。有一个与 TPU Ops here 相关的更一般的总结。

特别是，UpSampling2D 目前正在开发中，但如果您遇到更多有关 Keras 和 TPU 的问题，应鼓励您提交问题here。