Use general exception

apache · Jun 12, 2020 · 7d83a3f · 7d83a3f
1 parent d5848b6
commit 7d83a3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -1354,7 +1354,7 @@ def _impl(inputs, attr, params, mod):
         if output_shape is None or -1 in output_shape:
             try:
                 output_shape = _infer_value(inputs[0], params, mod).asnumpy().reshape([-1]).tolist()
-            except (IndexError, KeyError, AttributeError):
+            except Exception:
                 output_shape = inputs[0]
 
         dtype = attr['T'].name