diff --git a/experimental/torch_xla2/test/test_ops.py b/experimental/torch_xla2/test/test_ops.py index 28d0f29f0c1..848c55733e9 100644 --- a/experimental/torch_xla2/test/test_ops.py +++ b/experimental/torch_xla2/test/test_ops.py @@ -35,9 +35,6 @@ "nn.functional.conv_transpose2d", "nn.functional.conv_transpose3d", "nn.functional.ctc_loss", - "nn.functional.dropout2d", - "nn.functional.dropout3d", - "nn.functional.dropout", "nn.functional.embedding_bag", "nn.functional.fractional_max_pool2d", "nn.functional.fractional_max_pool3d", @@ -98,6 +95,9 @@ 'exponential', 'log_normal', 'randint', + 'nn.functional.dropout2d', + 'nn.functional.dropout3d', + 'nn.functional.dropout', } atol_dict = {"linalg.eig": (2e0, 3e0),