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.
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.
Copyright © 2021 Jogjafile Inc.
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.