Fix test failure.

Change-Id: Id7a7c31fca509507f7098097da8abc793506f833
This commit is contained in:
Elena Zhelezina 2020-08-24 08:26:28 +01:00
parent bf02b27dfc
commit 985adde321

View File

@ -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."""