Compilation failure: Detected unsupported operations when trying to compile graph get_loss_cond_1_true_88089_rewritten[]

Question

I\'m getting the following error on google colab TPU when try to use custom crf loss function. I check https://cloud.google.com/tpu/docs/tensorflow-ops for FakeParam operati