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

Question

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

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

Answer 1

通常应该支持所有 Keras 层，但没有一个很好的总结。这里有一个与 TPU Ops 相关的更一般的总结。

特别是，UpSampling2D 目前正在进行中，但如果您在使用 Keras 和 TPU 时遇到更多问题，应该鼓励您在此处提交问题。