resolve comments

Author: pengwa

tf2onnx/graph.py

@@ -19,8 +19,7 @@
 from onnx import helper, numpy_helper, optimizer, shape_inference, OperatorSetIdProto, AttributeProto
 from tf2onnx import utils, __version__
 from tf2onnx.utils import port_name, find_opset
-from tf2onnx.optimizer.identity_optimizer import IdentityOptimizer
-from tf2onnx.optimizer.transpose_optimizer import TransposeOptimizer
+from tf2onnx.optimizer import IdentityOptimizer, TransposeOptimizer
 from tf2onnx.schemas import get_schema
 
 

tf2onnx/optimizer/__init__.py

@@ -6,4 +6,10 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
-__all__ = ["identity_optimizer", "transpose_optimizer"]
+from .identity_optimizer import IdentityOptimizer
+from .transpose_optimizer import TransposeOptimizer
+
+__all__ = [
+    "IdentityOptimizer",
+    "TransposeOptimizer",
+]