Internal change.

PiperOrigin-RevId: 492558048

Author: fyangf

official/projects/qat/vision/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+

official/projects/qat/vision/configs/__init__.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """Configs package definition."""
-
+from official.projects.qat.vision.configs import common
 from official.projects.qat.vision.configs import image_classification
 from official.projects.qat.vision.configs import retinanet
 from official.projects.qat.vision.configs import semantic_segmentation

official/projects/qat/vision/modeling/__init__.py

@@ -13,5 +13,5 @@
 # limitations under the License.
 
 """Modeling package definition."""
-
+from official.projects.qat.vision.modeling import heads
 from official.projects.qat.vision.modeling import layers

official/projects/qat/vision/modeling/heads/__init__.py

@@ -12,7 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Lint as: python3
 """Heads package definition."""
-
 from official.projects.qat.vision.modeling.heads.dense_prediction_heads import RetinaNetHeadQuantized

official/projects/qat/vision/modeling/layers/__init__.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """Layers package definition."""
-
+from official.projects.qat.vision.modeling.layers import nn_layers
 from official.projects.qat.vision.modeling.layers.nn_blocks import BottleneckBlockQuantized
 from official.projects.qat.vision.modeling.layers.nn_blocks import Conv2DBNBlockQuantized
 from official.projects.qat.vision.modeling.layers.nn_blocks import InvertedBottleneckBlockQuantized

official/projects/qat/vision/serving/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+

official/projects/qat/vision/tasks/__init__.py

@@ -13,5 +13,6 @@
 # limitations under the License.
 
 """Tasks package definition."""
-
 from official.projects.qat.vision.tasks import image_classification
+from official.projects.qat.vision.tasks import retinanet
+from official.projects.qat.vision.tasks import semantic_segmentation