diff options
Diffstat (limited to 'tensorflow/python/grappler/model_analyzer.cc')
-rw-r--r-- | tensorflow/python/grappler/model_analyzer.cc | 9 |
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 << "]"; |