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

Ensure quantized dimension is 0 for conv and 3 for depthwise_conv as specified in https://www.tensorflow.org/lite/performance/quantization_spec

Browse Source 
PiperOrigin-RevId: 289966980
Change-Id: I2487e120e784aefd0d879be0616592ca0ecc45a6
This commit is contained in:
Nat Jeffries 2020-01-15 17:01:18 -08:00 committed by TensorFlower Gardener
parent b1b7f38c25
commit e23d46c234
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/micro/kernels/conv.cc
View File

@ -232,6 +232,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE(context, affine_quantization->zero_point);
// Conv is quantized along dimension 0:
// https://www.tensorflow.org/lite/performance/quantization_spec
TF_LITE_ENSURE_EQ(context, affine_quantization->quantized_dimension, 0);
TF_LITE_ENSURE_EQ(context, filter->dims->data[0],
affine_quantization->scale->size);
TF_LITE_ENSURE_EQ(context, filter->dims->data[0],

1
tensorflow/lite/micro/kernels/depthwise_conv.cc
View File

@ -224,6 +224,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE(context, affine_quantization->zero_point);
// Depthwise conv is quantized along dimension 3:
// https://www.tensorflow.org/lite/performance/quantization_spec
TF_LITE_ENSURE_EQ(context, affine_quantization->quantized_dimension, 3);
TF_LITE_ENSURE_EQ(context, filter->dims->data[3],
affine_quantization->scale->size);
TF_LITE_ENSURE_EQ(context, filter->dims->data[3],