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

Add fully quantize test for tranpose operator.

Browse Source 
PiperOrigin-RevId: 284047075
Change-Id: I83b211bc911567dae4a97c50817374b42052d13a
This commit is contained in:
A. Unique TensorFlower 2019-12-05 14:00:27 -08:00 committed by TensorFlower Gardener
parent 3cc04ef616
commit a3dc51f103
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/testing/BUILD
View File

@ -513,6 +513,7 @@ edgetpu_ops = [
"sub",
"sum", # high error
"tanh",
"transpose",
"transpose_conv",
]

18
tensorflow/lite/testing/op_tests/transpose.py
View File

@ -34,16 +34,31 @@ def make_transpose_tests(options):
"input_shape": [[2, 2, 3]],
"perm": [[0, 1, 2], [0, 2, 1]],
"constant_perm": [True, False],
"fully_quantize": [False],
}, {
"dtype": [tf.float32],
"input_shape": [[1, 2, 3, 4]],
"perm": [[0, 1, 2, 3], [3, 0, 1, 2]],
"constant_perm": [True, False],
"fully_quantize": [False],
}, {
"dtype": [tf.float32],
"input_shape": [[1, 2, 3, 4, 5]],
"perm": [[4, 3, 2, 1, 0]],
"constant_perm": [True, False],
"fully_quantize": [False],
}, {
"dtype": [tf.float32],
"input_shape": [[2, 2, 3]],
"perm": [[0, 1, 2], [0, 2, 1]],
"constant_perm": [True],
"fully_quantize": [True],
}, {
"dtype": [tf.float32],
"input_shape": [[1, 2, 3, 4]],
"perm": [[0, 1, 2, 3], [3, 0, 1, 2]],
"constant_perm": [True],
"fully_quantize": [True],
}]
def build_graph(parameters):
@ -66,7 +81,8 @@ def make_transpose_tests(options):
def build_inputs(parameters, sess, inputs, outputs):
values = [
create_tensor_data(parameters["dtype"], parameters["input_shape"])
create_tensor_data(parameters["dtype"], parameters["input_shape"],
min_value=-1, max_value=1)
]
if not parameters["constant_perm"]:
values.append(np.array(parameters["perm"]))