diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-02 15:16:03 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-02 15:20:45 -0700 |
| commit | bc2225a08ffb3714ddd8f391e1c731e4e7ca695e (patch) | |
| tree | 2f7a5a5742b93b6798494c549ceb0c905d7b4d75 /tensorflow/contrib/lite/toco/tooling_util.cc | |
| parent | 990677c71d07dace05f50afda918797af37c5fa1 (diff) | |
Fix array logging bug
PiperOrigin-RevId: 203021167
Diffstat (limited to 'tensorflow/contrib/lite/toco/tooling_util.cc')
| -rw-r--r-- | tensorflow/contrib/lite/toco/tooling_util.cc | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/tooling_util.cc b/tensorflow/contrib/lite/toco/tooling_util.cc index 7dc1af9f1d..01113506d0 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.cc +++ b/tensorflow/contrib/lite/toco/tooling_util.cc @@ -447,8 +447,12 @@ void LogSummary(int log_level, const Model& model) { } void LogArray(int log_level, const Model& model, const string& name) { - const auto& array = model.GetArray(name); VLOG(log_level) << "Array: " << name; + if (!model.HasArray(name)) { + VLOG(log_level) << " DOES NOT EXIST"; + return; + } + const auto& array = model.GetArray(name); VLOG(log_level) << " Data type: " << ArrayDataTypeName(array.data_type); VLOG(log_level) << " Final type: " << ArrayDataTypeName(array.final_data_type); |