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

Add dynamic range scheme to quantization testing

Browse Source 
PiperOrigin-RevId: 356175936
Change-Id: Ie9d5b45523ffcd27152f10da768a692c4ed20938
This commit is contained in:
Frederic Rechtenstein 2021-02-07 19:07:06 -08:00 committed by TensorFlower Gardener
parent 40fc414003
commit 93b319bd07
4 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/testing/model_coverage/model_coverage_quantization_test.py
View File

@ -39,6 +39,7 @@ NUM_CALIBRATION_STEPS = 100
@enum.unique
class QuantizationType(enum.Enum):
DYNAMIC_RANGE = "dynamicRange"
FLOAT16 = "float16"
FULL_INTEGER = "int"
FULL_INTEGER_16X8 = "int16x8"
@ -93,6 +94,7 @@ class ModelQuantizationTest(parameterized.TestCase):
converter.experimental_new_quantizer = True
converter.optimizations = [_lite.Optimize.DEFAULT]
# QuantizationType.DYNAMIC_RANGE: No additional converter option necessary
if quantization_type == QuantizationType.FLOAT16:
converter.target_spec.supported_types = [dtypes.float16]
elif quantization_type in (QuantizationType.FULL_INTEGER,

BIN
tensorflow/lite/testing/model_coverage/testdata/golden/inception_v3_dynamicRange.npy.golden vendored Normal file
View File

Binary file not shown.

BIN
tensorflow/lite/testing/model_coverage/testdata/golden/mobilenet_v1_dynamicRange.npy.golden vendored Normal file
View File

Binary file not shown.

BIN
tensorflow/lite/testing/model_coverage/testdata/golden/mobilenet_v2_dynamicRange.npy.golden vendored Normal file
View File

Binary file not shown.