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, 2 insertions, 7 deletions
diff --git a/tensorflow/python/grappler/model_analyzer.cc b/tensorflow/python/grappler/model_analyzer.cc
index 7d365c3be9..4ec7620bce 100644
--- a/tensorflow/python/grappler/model_analyzer.cc
+++ b/tensorflow/python/grappler/model_analyzer.cc
@@ -59,15 +59,10 @@ void ModelAnalyzer::PrintNodeInfo(const NodeDef* node,
if (i > 0) {
os << ", ";
}
- 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.
+ if (prop.shape().dim(i).size() < 0) {
os << "?";
} else {
- // Symbolic dimension.
- os << "x" << -prop.shape().dim(i).size();
+ os << prop.shape().dim(i).size();
}
}
os << "]";