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

Add quantize test for int16->int8.

Browse Source 
PiperOrigin-RevId: 346434549
Change-Id: Ia3cc28c6d1e463ab3466178d5714020054324ba3
This commit is contained in:
Nat Jeffries 2020-12-08 16:23:05 -08:00 committed by TensorFlower Gardener
parent 38b3c3c83d
commit 2a0fd3f6b1
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/lite/micro/kernels/quantize_test.cc
View File

@ -309,4 +309,23 @@ TF_LITE_MICRO_TEST(QuantizeOpTestInt16toInt32) {
output_zero_point, output_quantized); output_zero_point, output_quantized);
} }
TF_LITE_MICRO_TEST(QuantizeOpTestInt16toInt8) {
constexpr int length = 10;
const int dims[] = {2, 2, 5};
const float values[] = {-32, -31, -30, -29, -28, 27, 28, 29, 30, 31};
// TODO(b/155682734): Input scale must be smaller than output scale for
// xtensa.
const float input_scale = 0.4f;
const int input_zero_point = 0;
const float output_scale = 1.0f;
const int output_zero_point = 0;
int8_t output_quantized[length];
int8_t values_quantized[length];
int16_t input_quantized[length];
tflite::testing::TestRequantize(dims, values, input_quantized, input_scale,
input_zero_point, dims, values,
values_quantized, output_scale,
output_zero_point, output_quantized);
}
TF_LITE_MICRO_TESTS_END TF_LITE_MICRO_TESTS_END