diff options
author | 2018-02-20 13:00:26 -0800 | |
---|---|---|
committer | 2018-02-20 13:13:46 -0800 | |
commit | 6d1a1433707b37915207c11c2f0e91fcbc862bea (patch) | |
tree | 552b334c58bf3ba5780e1d7332596f244592c9d5 /tensorflow/contrib/lite/toco/dump_graphviz.cc | |
parent | 537166cf2bbb428fca1c5fda7a6ff157bbe5c44f (diff) |
Simplify and enforce diagnostic ArrayDataType strings.
PiperOrigin-RevId: 186348846
Diffstat (limited to 'tensorflow/contrib/lite/toco/dump_graphviz.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/dump_graphviz.cc | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/tensorflow/contrib/lite/toco/dump_graphviz.cc b/tensorflow/contrib/lite/toco/dump_graphviz.cc index c726eb6d86..2184e8f607 100644 --- a/tensorflow/contrib/lite/toco/dump_graphviz.cc +++ b/tensorflow/contrib/lite/toco/dump_graphviz.cc @@ -142,14 +142,8 @@ NodeProperties GetPropertiesForArray(const Model& model, // Append array shape to the label. auto& array = model.GetArray(array_name); - - if (array.data_type == ArrayDataType::kFloat) { - AppendF(&node_properties.label, "\\nType: float"); - } else if (array.data_type == ArrayDataType::kInt32) { - AppendF(&node_properties.label, "\\nType: int32"); - } else if (array.data_type == ArrayDataType::kUint8) { - AppendF(&node_properties.label, "\\nType: uint8"); - } + AppendF(&node_properties.label, "\\nType: %s", + ArrayDataTypeName(array.data_type)); if (array.has_shape()) { auto& array_shape = array.shape(); |