expose all stages in init (#302)

Summary:
Pull Request resolved: #302

as title

Reviewed By: digantdesai

Differential Revision: D48790083

fbshipit-source-id: ab9e5796fbba2cfc5eb8ea81bbf1d5d442012d0b

Author: mcr229

backends/xnnpack/test/models/mobilenet_v2.py

@@ -11,8 +11,7 @@
 from executorch.backends.xnnpack.partition.xnnpack_partitioner import (
     XnnpackQuantizedPartitioner,
 )
-from executorch.backends.xnnpack.test.tester import Partition, Tester
-from executorch.backends.xnnpack.test.tester.tester import Export
+from executorch.backends.xnnpack.test.tester import Export, Partition, Tester
 from executorch.exir import CaptureConfig
 from torchvision.models.mobilenetv2 import MobileNet_V2_Weights
 

backends/xnnpack/test/models/mobilenet_v3.py

@@ -11,8 +11,7 @@
 from executorch.backends.xnnpack.partition.xnnpack_partitioner import (
     XnnpackQuantizedPartitioner,
 )
-from executorch.backends.xnnpack.test.tester import Partition, Tester
-from executorch.backends.xnnpack.test.tester.tester import Export
+from executorch.backends.xnnpack.test.tester import Export, Partition, Tester
 from executorch.exir import CaptureConfig
 
 

backends/xnnpack/test/tester/__init__.py

@@ -5,6 +5,22 @@
 # LICENSE file in the root directory of this source tree.
 
 # TODO: Be more delibrate on module structure
-from executorch.backends.xnnpack.test.tester.tester import Partition, Tester
+from executorch.backends.xnnpack.test.tester.tester import (
+    Export,
+    Partition,
+    Quantize,
+    Serialize,
+    Tester,
+    ToEdge,
+    ToExecutorch,
+)
 
-__all__ = [Tester, Partition]
+__all__ = [
+    Tester,
+    Partition,
+    Quantize,
+    Export,
+    ToEdge,
+    ToExecutorch,
+    Serialize,
+]