diff --git a/test/test_transformers.py b/test/test_transformers.py index 7ac0e2da1f5d7a..af457828ebe17f 100644 --- a/test/test_transformers.py +++ b/test/test_transformers.py @@ -1,4 +1,4 @@ -# Owner(s): ["module: nn"] +# Owner(s): ["module: multi-headed-attention"] import contextlib from functools import partial