move function

Author: RandySheriffH

tf2onnx/custom_opsets/ms.py

@@ -92,3 +92,14 @@ def version_1(cls, ctx, node, **kwargs):
         node.attr.pop("padding")
         if "explicit_paddings" in node.attr:
             node.attr.pop("explicit_paddings")
+
+@tf_op("CropAndResize")
+class CropAndResize:
+    @classmethod
+    def version_11(cls, ctx, node, **kwargs):
+        """ utilize contrib cropandresize """
+        node.attr['method'].name = 'mode'
+        node.domain = constants.MICROSOFT_DOMAIN
+        ctx.insert_new_node_on_input(node, "Transpose", node.input[0], perm=constants.NHWC_TO_NCHW)
+        ctx.insert_new_node_on_output("Transpose", node.output[0], node.name + '_transposed',
+                                      None, perm=constants.NCHW_TO_NHWC)

tf2onnx/custom_opsets/onnx_ml.py

@@ -39,13 +39,3 @@ def version_8(cls, ctx, node, **kwargs):
         customer_nodes = ctx.find_output_consumers(table_node.output[0])
         if len(customer_nodes) == 0:
             ctx.remove_node(table_node.name)
-
-@tf_op(["CropAndResize"])
-class CropAndResize:
-    @classmethod
-    def version_11(cls, ctx, node, **kwargs):
-        """ utilize contrib cropandresize """
-        node.attr['method'].name = 'mode'
-        node.domain = constants.MICROSOFT_DOMAIN
-        ctx.insert_new_node_on_input(node, "Transpose", node.input[0], perm=constants.NHWC_TO_NCHW)
-        ctx.insert_new_node_on_output("Transpose", node.output[0], node.name + '_transposed', None, perm=constants.NCHW_TO_NHWC)