From 040aaf39aebda57921991d05d29be5123e908d7c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 14 May 2018 11:40:50 -0700 Subject: [PATCH] Don't check that bool arrays are quantized. PiperOrigin-RevId: 196541955 --- tensorflow/contrib/lite/toco/tooling_util.cc | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tensorflow/contrib/lite/toco/tooling_util.cc b/tensorflow/contrib/lite/toco/tooling_util.cc index 7a048f5..a789b5c 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.cc +++ b/tensorflow/contrib/lite/toco/tooling_util.cc @@ -2074,15 +2074,21 @@ bool ReshapeIsEquivalentToTranspose(const Model& model, void CheckFinalDataTypesSatisfied(const Model& model) { for (const auto& array_entry : model.GetArrayMap()) { const auto& array = *array_entry.second; + if (array.data_type == ArrayDataType::kBool) { + // Boolean values are never quantized. + continue; + } + // If the final data type is int16, the data type may be float, for example // after dequantization. if (array.final_data_type != ArrayDataType::kNone && array.final_data_type != ArrayDataType::kInt16) { - CHECK(array.final_data_type == array.data_type) + CHECK(array.data_type == array.final_data_type) << "Array \"" << array_entry.first - << "\" has mis-matching actual and final data types (" - << ArrayDataTypeName(array.data_type) << "," - << ArrayDataTypeName(array.final_data_type) << ")."; + << "\" has mis-matching actual and final data types (data_type=" + << ArrayDataTypeName(array.data_type) + << ", final_data_type=" << ArrayDataTypeName(array.final_data_type) + << ")."; } } } -- 2.7.4