add pt2tf tool

pt2tf/onnx-tensorflow/onnx_tf/gen_opset.py

@@ -0,0 +1,35 @@
+#!/usr/bin/env python
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import pprint
+
+from onnx import defs
+
+from onnx_tf.common.handler_helper import get_backend_coverage
+from onnx_tf.common.handler_helper import get_backend_partial_support_detail
+
+
+def main():
+  backend_opset_dict = {}
+
+  for schema in defs.get_all_schemas():
+    op_name = schema.name
+    backend_opset_dict[op_name] = []
+
+  backend_onnx_coverage, backend_experimental_op = get_backend_coverage()
+  backend_opset_dict.update(backend_onnx_coverage.get(defs.ONNX_DOMAIN, {}))
+  backend_ps_dict = get_backend_partial_support_detail()
+
+  with open('opset_version.py', 'w') as version_file:
+    pp = pprint.PrettyPrinter(indent=4)
+    version_file.write("backend_opset_version = {\n " +
+                       pp.pformat(backend_opset_dict)[1:-1] + "\n}\n\n")
+    version_file.write("backend_partial_support = {\n " +
+                       pp.pformat(backend_ps_dict)[1:-1] + "\n}\n")
+
+
+if __name__ == '__main__':
+  main()