Merge pull request #668 from PINTO0309/fix_dynamic_depth_to_space

Support for `DepthToSpace` transformation of dynamic tensors with multiple undefined dimensions

Author: PINTO0309

README.md

@@ -297,15 +297,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.25.1
+  ghcr.io/pinto0309/onnx2tf:1.25.2
 
   or
 
   # Authentication is not required for pulls from Docker Hub.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.25.1
+  docker.io/pinto0309/onnx2tf:1.25.2
 
   or
 

onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.25.1'
+__version__ = '1.25.2'

onnx2tf/ops/DepthToSpace.py

@@ -96,6 +96,14 @@ def make_node(
     elif mode == "CRD":
         batch, channel = input_tensor_shape[0], input_tensor_shape[-1]
         height, width = input_tensor_shape[1], input_tensor_shape[2]
+        if batch is None:
+            batch = tf.shape(input_tensor)[0]
+        if channel is None:
+            channel = tf.shape(input_tensor)[-1]
+        if height is None:
+            height = tf.shape(input_tensor)[1]
+        if width is None:
+            width = tf.shape(input_tensor)[2]
         csize = channel // (blocksize**2)
 
         reshape_node = tf.reshape(