From ea404dc688e27929093322a33477f1a3125b6a80 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 29 Oct 2019 14:16:29 -0700
Subject: [PATCH] Add mean operator to tflite int8 quantization test.

PiperOrigin-RevId: 277362640
Change-Id: I291a20210d7ce86603f95dce9c1feaf64e9aee0d
---
 tensorflow/lite/testing/BUILD              | 1 +
 tensorflow/lite/testing/op_tests/reduce.py | 8 +++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

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()