From 712fc6252228748a72bbc015be55bd20ba811cbb Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 4 Apr 2018 13:38:20 -0700 Subject: [PATCH] Permit use of ArraysExtraInfo/constant_float_value when target model is quantized. PiperOrigin-RevId: 191639289 --- tensorflow/contrib/lite/toco/tooling_util.cc | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tensorflow/contrib/lite/toco/tooling_util.cc b/tensorflow/contrib/lite/toco/tooling_util.cc index 668cf51..56fa8f4 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.cc +++ b/tensorflow/contrib/lite/toco/tooling_util.cc @@ -2027,11 +2027,12 @@ void UseArraysExtraInfo(Model* model) { } if (entry.has_constant_float_value()) { CHECK(array.has_shape()); - CHECK(array.data_type == ArrayDataType::kFloat); - auto& data = array.GetMutableBuffer().data; - data.resize(RequiredBufferSizeForShape(array.shape())); - for (float& f : data) { - f = entry.constant_float_value(); + if (array.data_type == ArrayDataType::kFloat) { + auto& data = array.GetMutableBuffer().data; + data.resize(RequiredBufferSizeForShape(array.shape())); + for (float& f : data) { + f = entry.constant_float_value(); + } } } } -- 2.7.4