Merge pull request #621 from PINTO0309/pixcel_unshuffle

Optimization of `torch.nn.PixelUnshuffle`

README.md

@@ -270,15 +270,15 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  ghcr.io/pinto0309/onnx2tf:1.20.6
+  ghcr.io/pinto0309/onnx2tf:1.20.7
 
   or
 
   # Authentication is not required for pulls from Docker Hub.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.20.6
+  docker.io/pinto0309/onnx2tf:1.20.7
 
   or
 

onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.20.6'
+__version__ = '1.20.7'

onnx2tf/onnx2tf.py

@@ -820,6 +820,7 @@ def sanitizing(node):
         'output_nms_with_dynamic_tensor': output_nms_with_dynamic_tensor,
         'output_integer_quantized_tflite': output_integer_quantized_tflite,
         'gelu_replace_op_names': {},
+        'space_to_depth_replace_op_names': {},
         'relu_relu6_merge_op_names': {},
         'mul_div_replace_op_names': {},
         'use_cuda': use_cuda,

onnx2tf/ops/ReduceL1.py

@@ -162,7 +162,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceL2.py

@@ -162,7 +162,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceLogSum.py

@@ -162,7 +162,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceLogSumExp.py

@@ -162,7 +162,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceMax.py

@@ -169,7 +169,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceMean.py

@@ -163,7 +163,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceMin.py

@@ -163,7 +163,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceProd.py

@@ -163,7 +163,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceSum.py

@@ -162,7 +162,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize

onnx2tf/ops/ReduceSumSquare.py

@@ -163,7 +163,7 @@ def make_node(
                 del onnx_tensor_infos_for_validation
 
     if not disable_strict_mode:
-        if onnx_tensor_infos is not None and validation_data is not None:
+        if onnx_tensor_infos is not None and validation_data is not None and axes is not None:
             # Shape Unmatch Error Mitigation Measures
             # Search for and transpose shapes that do not cause shape unmatch errors
             min_abs_err = sys.maxsize