diff options
Diffstat (limited to 'tensorflow/contrib/graph_editor')
-rw-r--r-- | tensorflow/contrib/graph_editor/util.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/graph_editor/util.py b/tensorflow/contrib/graph_editor/util.py index 959905e982..30bc33b9ee 100644 --- a/tensorflow/contrib/graph_editor/util.py +++ b/tensorflow/contrib/graph_editor/util.py @@ -93,6 +93,8 @@ class ListView(object): # TODO(fkp): very generic code, it should be moved in a more generic place. def is_iterable(obj): """Return true if the object is iterable.""" + if isinstance(obj, tf_ops.Tensor): + return False try: _ = iter(obj) except Exception: # pylint: disable=broad-except |