1

I recently learned that UpSampling2D is not a TPU supported operation.

How do I find out which Keras functions are compatible with a TPU in Colab? I was unable to find any documentation on the subject.

James Kl
  • 177
  • 9
  • Possible duplicate of [Keras TPU. Compilation failure: Detected unsupported operations](https://stackoverflow.com/questions/52906186/keras-tpu-compilation-failure-detected-unsupported-operations) – gonadarian Jan 03 '19 at 16:37

1 Answers1

0

Generally all Keras layers should be supported, but there is not a nice summary. There is a more general summary that is related to TPU Ops here.

In particular, UpSampling2D is a work in progress at this time, but you should be encouraged to file issues here if you are having more problems with Keras and TPUs.

Alex Ilchenko
  • 322
  • 1
  • 3