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

Remove TFLite reduce test cases with redundant axes

Browse Source 
PiperOrigin-RevId: 318844070
Change-Id: If4b385d4988ca71808dc8c94c617aaf6a406f260
This commit is contained in:
Jared Duke 2020-06-29 10:37:39 -07:00 committed by TensorFlower Gardener
parent 177f89fb3c
commit e3036c5cc9
1 changed files with 60 additions and 11 deletions

71
tensorflow/lite/testing/op_tests/reduce.py
View File

@ -50,9 +50,25 @@ def make_reduce_tests(reduce_op,
"input_dtype": [tf.float32, tf.int32, tf.int64],
"input_shape": [[3, 3, 2, 4]],
"axis": [
0, 1, 2, [0, 1], [0, 2], [1, 2], [0, 1, 2], [1, 0], [2, 0],
[2, 1], [2, 1, 0], [2, 0, 1], -1, -2, -3, [1, -1], [0, -1],
[-1, 0], [-1, -2, -3], [0, 0, 0], [2, 2, 0], [1, 0, -3, -3]
0,
1,
2,
[0, 1],
[0, 2],
[1, 2],
[0, 1, 2],
[1, 0],
[2, 0],
[2, 1],
[2, 1, 0],
[2, 0, 1],
-1,
-2,
-3,
[1, -1],
[0, -1],
[-1, 0],
[-1, -2, -3],
],
"const_axis": [True, False],
"keepdims": [True, False],
@ -62,11 +78,28 @@ def make_reduce_tests(reduce_op,
"input_dtype": [tf.float32],
"input_shape": [[1, 8, 8, 3]],
"axis": [
0, 1, 2, 3, [1, 2], [0, 3], [1, 2, 3], [0, 1, 2,
3], [3, 2, 1, 0],
[3, 1, 0, 2], [2, 0], [3, 0], [3, 1], [1, 0], -1, -2, -3, -4,
[0, -2], [2, 3, -1, 0], [3, 1, 2, -3], [3, -4], [2, 2, 2],
[2, 2, 3], [-3, -3, -4], [-3, 2, 1]
0,
1,
2,
3,
[1, 2],
[0, 3],
[1, 2, 3],
[0, 1, 2, 3],
[3, 2, 1, 0],
[3, 1, 0, 2],
[2, 0],
[3, 0],
[3, 1],
[1, 0],
-1,
-2,
-3,
-4,
[0, -2],
[2, 3, 1, 0],
[3, 1, 2],
[3, -4],
],
"const_axis": [True, False],
"keepdims": [True, False],
@ -92,9 +125,25 @@ def make_reduce_tests(reduce_op,
"input_dtype": [tf.float32],
"input_shape": [[3, 3, 2, 4]],
"axis": [
0, 1, 2, [0, 1], [0, 2], [1, 2], [0, 1, 2], [1, 0], [2, 0],
[2, 1], [2, 1, 0], [2, 0, 1], -1, -2, -3, [1, -1], [0, -1],
[-1, 0], [-1, -2, -3], [0, 0, 0], [2, 2, 0], [1, 0, -3, -3]
0,
1,
2,
[0, 1],
[0, 2],
[1, 2],
[0, 1, 2],
[1, 0],
[2, 0],
[2, 1],
[2, 1, 0],
[2, 0, 1],
-1,
-2,
-3,
[1, -1],
[0, -1],
[-1, 0],
[-1, -2, -3],
],
"const_axis": [True],
"keepdims": [True, False],