TPU in Colab is not the same device as edge TPU.