AveragePool bug fixes

PINTO0309 · Oct 29, 2022 · d438145 · d438145
1 parent 578cdd0
commit d438145
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -71,7 +71,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.0.31
+ghcr.io/pinto0309/onnx2tf:1.0.32
 
 or
 

diff --git a/onnx2tf/__init__.py b/onnx2tf/__init__.py
@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.0.31'
+__version__ = '1.0.32'
diff --git a/onnx2tf/ops/AveragePool.py b/onnx2tf/ops/AveragePool.py
@@ -63,7 +63,7 @@ def make_node(
     x_rank = spatial_size + 2
     strides = graph_node.attrs.get('strides', [1] * spatial_size)
     dilations = graph_node.attrs.get('dilations', [1] * spatial_size)
-    is_known_shape = input_tensor.shape.is_fully_defined()
+    is_known_shape = None not in input_tensor.shape
 
     pads = graph_node.attrs.get('auto_pad', 'NOTSET')
     if pads == 'NOTSET':
@@ -76,6 +76,7 @@ def make_node(
                 strides=strides,
                 dilations=dilations,
                 padding='SAME_UPPER',
+                is_known_shape=is_known_shape,
             )
             if pads == same_paddings:
                 pads = 'SAME_UPPER'