raise right error in tensorflow split op (apache#5951)

neo-ai · Jul 2, 2020 · 28fa03d · 28fa03d
1 parent 34ce297
commit 28fa03d
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
@@ -1742,7 +1742,7 @@ def _impl(inputs, attr, params, mod):
                 indices_or_sections = attr['num_split']
             input_node = inputs[input_node_index]
             axis_input_value = _get_num_param(params, inputs[input_axis_index])
-        except (IndexError, KeyError):
+        except (IndexError, KeyError, AttributeError):
             raise TypeError(
                 "Unsupported argument for split: `axis` and `num_or_size_splits` "
                 "should be constants")