diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
index 107d8df9cb6f1..0b037a9fc05bd 100644
--- 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