[incubator-tvm] branch master updated: raise right error in tensorflow split op (#5951)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 2b3d90a  raise right error in tensorflow split op (#5951)
2b3d90a is described below

commit 2b3d90ab9b8e57fb3381c43b109b41f20fa97eca
Author: Meteorix <lx...@gmail.com>
AuthorDate: Mon Jun 29 00:28:15 2020 +0800

    raise right error in tensorflow split op (#5951)
---
 python/tvm/relay/frontend/tensorflow.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
index 62dadce..b5f34e4 100644
--- a/python/tvm/relay/frontend/tensorflow.py
+++ b/python/tvm/relay/frontend/tensorflow.py
@@ -1742,7 +1742,7 @@ def _split(has_size_vector):
                 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")