onnx2tf 1.27.1__tar.gz → 1.27.2__tar.gz

{onnx2tf-1.27.1/onnx2tf.egg-info → onnx2tf-1.27.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnx2tf
-Version: 1.27.1
+Version: 1.27.2
 Summary: Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).
 Home-page: https://github.com/PINTO0309/onnx2tf
 Author: Katsuya Hyodo
@@ -334,7 +334,7 @@ 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.27.1
+  ghcr.io/pinto0309/onnx2tf:1.27.2
 
   or
 
@@ -342,7 +342,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.27.1
+  docker.io/pinto0309/onnx2tf:1.27.2
 
   or
 

{onnx2tf-1.27.1 → onnx2tf-1.27.2}/README.md

@@ -309,7 +309,7 @@ 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.27.1
+  ghcr.io/pinto0309/onnx2tf:1.27.2
 
   or
 
@@ -317,7 +317,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.27.1
+  docker.io/pinto0309/onnx2tf:1.27.2
 
   or
 

{onnx2tf-1.27.1 → onnx2tf-1.27.2}/onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.27.1'
+__version__ = '1.27.2'

{onnx2tf-1.27.1 → onnx2tf-1.27.2}/onnx2tf/ops/Split.py

@@ -158,15 +158,56 @@ def make_node(
                 if idx == axis:
                     end_.append(split_idx + 1)
                 elif input_tensor_shape[idx] is None:
-                    end_.append(-1)
+                    end_.append(0)
                 else:
                     end_.append(input_tensor_shape[idx])
 
+            begin_mask_ = np.sum([2**idx if idx != axis else 0 for idx in range(input_tensor_rank)])
+            end_mask_ = np.sum([2**idx if idx != axis else 0 for idx in range(input_tensor_rank)])
+
             splited_tensors.append(
                 tf.strided_slice(
                     input_=input_tensor,
                     begin=begin_,
                     end=end_,
+                    begin_mask=begin_mask_,
+                    end_mask=end_mask_,
+                )
+            )
+    elif isinstance(split, np.ndarray) \
+        and len(list(split)) > 1 \
+        and np.prod(split) != 1 \
+        and np.all(split == split[0]) \
+        and isinstance(input_tensor_shape[axis], int) \
+        and input_tensor_shape[axis] == np.sum(split):
+        # strided_slice - Slice everything in same size
+        # Suppression of FlexSplitV generation
+        # https://github.com/PINTO0309/onnx2tf/issues/751
+        splited_tensors = []
+        split_size = split[0]
+        for split_idx in range(len(list(split))):
+            begin_ = [
+                split_size * split_idx if idx == axis else 0 for idx in range(input_tensor_rank)
+            ]
+            end_ = []
+            for idx in range(input_tensor_rank):
+                if idx == axis:
+                    end_.append(split_size * split_idx + split_size)
+                elif input_tensor_shape[idx] is None:
+                    end_.append(0)
+                else:
+                    end_.append(input_tensor_shape[idx])
+
+            begin_mask_ = np.sum([2**idx if idx != axis else 0 for idx in range(input_tensor_rank)])
+            end_mask_ = np.sum([2**idx if idx != axis else 0 for idx in range(input_tensor_rank)])
+
+            splited_tensors.append(
+                tf.strided_slice(
+                    input_=input_tensor,
+                    begin=begin_,
+                    end=end_,
+                    begin_mask=begin_mask_,
+                    end_mask=end_mask_,
                 )
             )
     elif isinstance(split, np.ndarray) \

{onnx2tf-1.27.1 → onnx2tf-1.27.2/onnx2tf.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnx2tf
-Version: 1.27.1
+Version: 1.27.2
 Summary: Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).
 Home-page: https://github.com/PINTO0309/onnx2tf
 Author: Katsuya Hyodo
@@ -334,7 +334,7 @@ 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.27.1
+  ghcr.io/pinto0309/onnx2tf:1.27.2
 
   or
 
@@ -342,7 +342,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.27.1
+  docker.io/pinto0309/onnx2tf:1.27.2
 
   or