experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tflite] Fix OSS build: int32 --> int32_t

Browse Source 
PiperOrigin-RevId: 283797658
Change-Id: I1ff08717dbc83f88cd3aed3b413c3a482bc904e5
This commit is contained in:
Shanqing Cai 2019-12-04 11:30:12 -08:00 committed by TensorFlower Gardener
parent 5cb564092f
commit 8ba0ba1ea2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/testing/tflite_driver.cc
View File

@ -256,11 +256,11 @@ bool TfLiteDriver::DataExpectation::QuantizedCheck(bool verbose,
auto* quantization = auto* quantization =
reinterpret_cast<TfLiteAffineQuantization*>(tensor.quantization.params); reinterpret_cast<TfLiteAffineQuantization*>(tensor.quantization.params);
const float scale = quantization->scale->data[0]; const float scale = quantization->scale->data[0];
const int32 zero_point = quantization->zero_point->data[0]; const int32_t zero_point = quantization->zero_point->data[0];
bool good_result = true; bool good_result = true;
for (int i = 0; i < tensor.bytes; i++) { for (int i = 0; i < tensor.bytes; i++) {
const int32 computed = tensor.data.int8[i]; const int32_t computed = tensor.data.int8[i];
const float dequantized = const float dequantized =
static_cast<float>(scale * (computed - zero_point)); static_cast<float>(scale * (computed - zero_point));
const float reference = Value<float>(data_.get(), i); const float reference = Value<float>(data_.get(), i);