Add mean operator to tflite int8 quantization test.

PiperOrigin-RevId: 277362640
Change-Id: I291a20210d7ce86603f95dce9c1feaf64e9aee0d
This commit is contained in:
A. Unique TensorFlower 2019-10-29 14:16:29 -07:00 committed by TensorFlower Gardener
parent d7a15a47eb
commit ea404dc688
2 changed files with 8 additions and 1 deletions

View File

@ -491,6 +491,7 @@ edgetpu_ops = [
"fully_connected",
"l2norm", # high error
"max_pool",
"mean",
"mul",
"pad", # high error
"relu6",

View File

@ -169,7 +169,13 @@ def make_reduce_tests(reduce_op,
@register_make_test_function()
def make_mean_tests(options):
"""Make a set of tests to do mean."""
return make_reduce_tests(tf.reduce_mean)(options)
return make_reduce_tests(
tf.reduce_mean,
min_value=-1,
max_value=1,
boolean_tensor_only=False,
allow_fully_quantize=True)(
options)
@register_make_test_function()