reorg habndlers tfonnx.py into category based files

Author: guschmue

tests/run_pretrained_models.yaml

@@ -185,17 +185,6 @@ mobilenet_v1_75_192:
   outputs:
     - MobilenetV1/Predictions/Softmax:0
 
-tiny-yolo:
-  # works but local file
-  disabled: true
-  model: c:/src/darkflow/built_graph/tiny-yolo.pb
-  input_get: get_beach
-  inputs:
-    "input:0": [1, 416, 416, 3]
-  outputs:
-    - output:0
-  rtol: 0.6
-
 nasnet-a_mobile_224:
   # has only checkpoint format
   disabled: true

tf2onnx/constants.py

@@ -27,3 +27,8 @@
 TARGET_CAFFE2 = "caffe2"
 POSSIBLE_TARGETS = [TARGET_RS4, TARGET_RS5, TARGET_RS6, TARGET_CAFFE2]
 DEFAULT_TARGET = []
+
+NCHW_TO_NHWC = [0, 2, 3, 1]
+NHWC_TO_NCHW = [0, 3, 1, 2]
+HWCN_TO_NCHW = [3, 2, 0, 1]
+NCHW_TO_HWCN = [2, 3, 1, 0]

tf2onnx/custom_opsets/__init__.py

@@ -4,7 +4,3 @@
 
 from . import ms
 from .. import constants
-
-DOMAIN_OPSETS = {
-    constants.MICROSOFT_DOMAIN: ms.OPSETS
-}

tf2onnx/custom_opsets/ms.py

@@ -4,24 +4,15 @@
 
 from onnx.onnx_pb import TensorProto
 from tf2onnx import constants, utils
-from tf2onnx.function.range import make_range_const
+from tf2onnx.handler import tf_op
+from tf2onnx.onnx_opset import controlflow
 
 
-# pylint: disable=unused-argument
-
-def range_op1(ctx, node, name, args):
-    """Range."""
-    # T range = Range(T start, T limit, T delta)
-    dtype = node.get_attr_int("Tidx")
-    shape = node.output_shapes[0]
-    utils.make_sure(dtype is not None, "Tidx of %s is None", node.name)
-    ctx.remove_node(node.name)
-    make_range(ctx, node.input[0], node.input[1], node.input[2], node.output[0], name, shape, dtype)
-
+# pylint: disable=unused-argument,missing-docstring
 
 def make_range(ctx, start, limit, delta, output, scope_name, shape, dtype):
     if all(ctx.get_node_by_output(n).is_const() for n in [start, limit, delta]) is True:
-        make_range_const(ctx, start, limit, delta, output, scope_name, shape, dtype)
+        controlflow.make_range_const(ctx, start, limit, delta, output, scope_name, shape, dtype)
     else:
         _make_range_non_const(ctx, start, limit, delta, output, scope_name, shape, dtype)
 
@@ -34,10 +25,14 @@ def _make_range_non_const(ctx, start, limit, delta, output, scope_name, shape, d
                   domain=constants.MICROSOFT_DOMAIN)
 
 
-_OPSET_1 = {
-    "Range": (range_op1, []),
-}
-
-OPSETS = [
-    (1, _OPSET_1),
-]
+@tf_op("Range", domain=constants.MICROSOFT_DOMAIN)
+class Range:
+    @classmethod
+    def version_1(cls, ctx, node, **kwargs):
+        """Range."""
+        # T range = Range(T start, T limit, T delta)
+        dtype = node.get_attr_int("Tidx")
+        shape = node.output_shapes[0]
+        utils.make_sure(dtype is not None, "Tidx of %s is None", node.name)
+        ctx.remove_node(node.name)
+        make_range(ctx, node.input[0], node.input[1], node.input[2], node.output[0], node.name, shape, dtype)