diff options
author | 2017-11-09 16:45:14 -0800 | |
---|---|---|
committer | 2017-11-10 16:14:41 -0800 | |
commit | 4506cb5ce176cb16bfed9a2b460d6392af43c0d6 (patch) | |
tree | f504b51a86bc695134bf9ff75a588e8a24a56fd3 | |
parent | be4b984747402c224d9aa3687e94db3da972ecf4 (diff) |
Improved the reporting of dimensions
PiperOrigin-RevId: 175232587
-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 << "]"; |