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