diff --git a/tensorflow/lite/testing/op_tests/leaky_relu.py b/tensorflow/lite/testing/op_tests/leaky_relu.py index bfd03d0ebfd..72f7c5b94b3 100644 --- a/tensorflow/lite/testing/op_tests/leaky_relu.py +++ b/tensorflow/lite/testing/op_tests/leaky_relu.py @@ -32,15 +32,11 @@ def make_leaky_relu_tests(options): { "input_shape": [[], [1], [5], [1, 10, 10, 3], [3, 3, 3, 3]], "alpha": [0.1, 1.0, 2.0, -0.1, -1.0, -2.0], - "fully_quantize": [False], - "quant_16x8": [False], - }, - { - "input_shape": [[1, 10, 10, 3]], - "alpha": [0.1], - "fully_quantize": [True], - "quant_16x8": [True, False], - }] + "fully_quantize": [False, True], + "input_range": [(-3, 10)], + "quant_16x8": [False, True], + } + ] def build_graph(parameters): """Build the graph for the test case."""