diff options
-rw-r--r-- | tensorflow/contrib/lite/tools/visualize.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/tools/visualize.py b/tensorflow/contrib/lite/tools/visualize.py index 597dede63b..d7eea79399 100644 --- a/tensorflow/contrib/lite/tools/visualize.py +++ b/tensorflow/contrib/lite/tools/visualize.py @@ -202,7 +202,7 @@ class TensorMapper(object): html += str(i) + " " html += tensor["name"] + " " html += str(tensor["type"]) + " " - html += repr(tensor["shape"]) + "<br>" + html += (repr(tensor["shape"]) if "shape" in tensor else "[]") + "<br>" html += "</span>" html += repr(x) html += "</span>" |