diff --git a/tensorflow_models/vision/__init__.py b/tensorflow_models/vision/__init__.py
index 6057d9054b3..927909837eb 100644
--- a/tensorflow_models/vision/__init__.py
+++ b/tensorflow_models/vision/__init__.py
@@ -16,3 +16,4 @@
 from official.vision import configs
 from official.vision import serving
 from official.vision.modeling import *
+from official.vision.ops import *