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

Addressed reviewer's comments. 

Change-Id: I4b849e60540879ca89483ede675c63631bc9417b
This commit is contained in:
Elena Zhelezina 2020-06-17 12:22:53 +01:00
parent 01b30fa03f
commit 286cd7fc68
1 changed files with 7 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tensorflow/lite/python/lite.py
View File

@ -299,32 +299,21 @@ class QuantizationMode(object):
inference_input_type = input_ty if input_ty else constants.FLOAT
inference_output_type = output_ty if output_ty else constants.FLOAT
if self.post_training_int8_no_float():
if self.post_training_int8_no_float() \
or self.post_training_int16x8_no_float():
return True, {
"inference_input_type": inference_input_type,
"inference_output_type": inference_output_type,
"activations_type": constants.INT8,
"activations_type": self.activations_type(),
"allow_float": False
}
elif self.post_training_int8_allow_float():
elif self.post_training_int8_allow_float() \
or self.post_training_int16x8_allow_float():
return True, {
"inference_input_type": inference_input_type,
"inference_output_type": inference_output_type,
"activations_type": constants.INT8,
"allow_float": True
}
elif self.post_training_int16x8_no_float():
return True, {
"inference_input_type": inference_input_type,
"inference_output_type": inference_output_type,
"activations_type": constants.INT16,
"allow_float": False
}
elif self.post_training_int16x8_allow_float():
return True, {
"inference_input_type": inference_input_type,
"inference_output_type": inference_output_type,
"activations_type": constants.INT16,
"activations_type": self.activations_type(),
"allow_float": True
}
else: