experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #41342 from wwwind:16x8_op_tests_leaky_relu

Browse Source 
PiperOrigin-RevId: 328692051
This commit is contained in:
TensorFlower Gardener 2020-08-27 01:11:07 -07:00
commit 25636f8bcb
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/lite/testing/op_tests/leaky_relu.py
View File

@ -28,12 +28,13 @@ from tensorflow.lite.testing.zip_test_utils import register_make_test_function
def make_leaky_relu_tests(options): def make_leaky_relu_tests(options):
"""Make a set of tests to do LeakyRelu.""" """Make a set of tests to do LeakyRelu."""
test_parameters = [ test_parameters = [{
{ "input_shape": [[], [1], [5], [1, 10, 10, 3], [3, 3, 3, 3]],
"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],
"alpha": [0.1, 1.0, 2.0, -0.1, -1.0, -2.0], "fully_quantize": [False, True],
}, "input_range": [(-3, 10)],
] "quant_16x8": [False, True],
}]
def build_graph(parameters): def build_graph(parameters):
"""Build the graph for the test case.""" """Build the graph for the test case."""