Add mean operator to tflite int8 quantization test.
PiperOrigin-RevId: 277362640 Change-Id: I291a20210d7ce86603f95dce9c1feaf64e9aee0d
This commit is contained in:
parent
d7a15a47eb
commit
ea404dc688
@ -491,6 +491,7 @@ edgetpu_ops = [
|
||||
"fully_connected",
|
||||
"l2norm", # high error
|
||||
"max_pool",
|
||||
"mean",
|
||||
"mul",
|
||||
"pad", # high error
|
||||
"relu6",
|
||||
|
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user