Where can I identify which Keras methods are TPU compatible methods?

349 Views Asked by At

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.

1

There are 1 best solutions below

0
On BEST ANSWER

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.