Add XLA test (#9848)

2021-01-29 11:25:03 +01:00
parent 99b9affa02
commit fdcde144d8
18 changed files with 85 additions and 0 deletions
--- a/tests/test_modeling_tf_convbert.py
+++ b/tests/test_modeling_tf_convbert.py
@@ -301,6 +301,10 @@ class TFConvBertModelTest(TFModelTesterMixin, unittest.TestCase):
                    [self.model_tester.num_attention_heads / 2, encoder_seq_length, encoder_key_length],
                )

+    def test_xla_mode(self):
+        # TODO JP: Make ConvBert XLA compliant
+        pass
+
    @slow
    def test_model_from_pretrained(self):
        model = TFConvBertModel.from_pretrained("YituTech/conv-bert-base")