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

Add test examples for SQUEEZE when leading number is not 1.

Browse Source 
PiperOrigin-RevId: 293472803
Change-Id: I1140a50596ef8f24d56ac769ae825116062efa7f
This commit is contained in:
A. Unique TensorFlower 2020-02-05 15:59:34 -08:00 committed by TensorFlower Gardener
parent 0f27efc0f8
commit b7907e9465
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/testing/op_tests/squeeze.py
View File

@ -62,8 +62,8 @@ def make_squeeze_tests(options):
"fully_quantize": [True], "fully_quantize": [True],
}, { }, {
"dtype": [tf.float32], "dtype": [tf.float32],
"input_shape": [[1, 1, 5, 10], [1, 5, 1, 10]], "input_shape": [[1, 1, 5, 10], [1, 5, 1, 10], [5, 1, 10]],
"axis": [[0], [3, 0], [-2, 0, 3, 2]], "axis": [[0], [1], [3, 0], [-2, 0, 3, 2]],
"fully_quantize": [True], "fully_quantize": [True],
}] }]
@ -89,4 +89,4 @@ def make_squeeze_tests(options):
test_parameters, test_parameters,
build_graph, build_graph,
build_inputs, build_inputs,
expected_tf_failures=20) expected_tf_failures=24)