aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/grappler/model_analyzer.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/grappler/model_analyzer.cc')
-rw-r--r--tensorflow/python/grappler/model_analyzer.cc9
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/python/grappler/model_analyzer.cc b/tensorflow/python/grappler/model_analyzer.cc
index 4ec7620bce..7d365c3be9 100644
--- a/tensorflow/python/grappler/model_analyzer.cc
+++ b/tensorflow/python/grappler/model_analyzer.cc
@@ -59,10 +59,15 @@ void ModelAnalyzer::PrintNodeInfo(const NodeDef* node,
if (i > 0) {
os << ", ";
}
- if (prop.shape().dim(i).size() < 0) {
+ if (prop.shape().dim(i).size() >= 0) {
+ // Print the actual dimension.
+ os << prop.shape().dim(i).size();
+ } else if (prop.shape().dim(i).size() == -1) {
+ // We don't know anything about the dimension.
os << "?";
} else {
- os << prop.shape().dim(i).size();
+ // Symbolic dimension.
+ os << "x" << -prop.shape().dim(i).size();
}
}
os << "]";