diff --git a/tensorflow/lite/testing/BUILD b/tensorflow/lite/testing/BUILD index 769d17ce703..395ee8755be 100644 --- a/tensorflow/lite/testing/BUILD +++ b/tensorflow/lite/testing/BUILD @@ -491,6 +491,7 @@ edgetpu_ops = [ "fully_connected", "l2norm", # high error "max_pool", + "mean", "mul", "pad", # high error "relu6", diff --git a/tensorflow/lite/testing/op_tests/reduce.py b/tensorflow/lite/testing/op_tests/reduce.py index 17be06bff4a..ae4513ef952 100644 --- a/tensorflow/lite/testing/op_tests/reduce.py +++ b/tensorflow/lite/testing/op_tests/reduce.py @@ -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()