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

Add generated tests for int8 type for mirror_pad

Browse Source 
PiperOrigin-RevId: 306527646
Change-Id: I8d5582f12097ef94d0d9d8640f5a5b2cb48df3d3
This commit is contained in:
Karim Nosir 2020-04-14 15:32:02 -07:00 committed by TensorFlower Gardener
parent 5394d6e97b
commit 07406a0ead
1 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tensorflow/lite/testing/op_tests/mirror_pad.py
View File

@ -33,49 +33,57 @@ def make_mirror_pad_tests(options):
"input_shape": [[2, 3]],
"padding_matrix": [[[1, 1], [2, 1]]],
"mode": ["REFLECT"],
"type": ["const"]
"type": ["const"],
"fully_quantize": [True, False],
},
{
"input_shape": [[2, 3]],
"padding_matrix": [[[1, 1], [1, 1]]],
"mode": ["REFLECT"],
"type": ["const"]
"type": ["const"],
"fully_quantize": [False],
},
{
"input_shape": [[2, 3]],
"padding_matrix": [[[1, 1], [2, 1]]],
"mode": ["SYMMETRIC"],
"type": ["placeholder"]
"type": ["placeholder"],
"fully_quantize": [False],
},
{
"input_shape": [[2, 3]],
"padding_matrix": [[[1, 1], [2, 1]]],
"mode": ["REFLECT"],
"type": ["placeholder"]
"type": ["placeholder"],
"fully_quantize": [False],
},
{
"input_shape": [[3]],
"padding_matrix": [[[0, 2]]],
"mode": ["SYMMETRIC"],
"type": ["placeholder"]
"type": ["placeholder"],
"fully_quantize": [False],
},
{
"input_shape": [[3]],
"padding_matrix": [[[0, 2]]],
"mode": ["SYMMETRIC"],
"type": ["const"]
"type": ["const"],
"fully_quantize": [False],
},
{
"input_shape": [[3]],
"padding_matrix": [[[0, 2]]],
"mode": ["REFLECT"],
"type": ["const"]
"type": ["const"],
"fully_quantize": [False, True],
},
{
"input_shape": [[3, 2, 4, 5]],
"padding_matrix": [[[1, 1], [2, 2], [1, 1], [1, 1]]],
"mode": ["SYMMETRIC"],
"type": ["placeholder"]
"type": ["placeholder"],
"fully_quantize": [False],
},
]
@ -83,8 +91,8 @@ def make_mirror_pad_tests(options):
"""Build the graph for the test case."""
input_tensor = tf.compat.v1.placeholder(
dtype=tf.int32, name="input", shape=parameters["input_shape"])
if parameters["type"] != "const":
dtype=tf.float32, name="input", shape=parameters["input_shape"])
if parameters["type"] != "const" and not parameters["fully_quantize"]:
padding_matrix = tf.compat.v1.placeholder(
dtype=tf.int32,
name="padding",
@ -99,7 +107,13 @@ def make_mirror_pad_tests(options):
return input_tensors, [output]
def build_inputs(parameters, sess, inputs, outputs):
input_values = [create_tensor_data(tf.int32, parameters["input_shape"])]
if not parameters["fully_quantize"]:
input_values = [create_tensor_data(tf.float32, parameters["input_shape"])]
else:
input_values = [
create_tensor_data(
tf.float32, parameters["input_shape"], min_value=-1, max_value=1)
]
if parameters["type"] != "const":
input_values.append(np.array(parameters["padding_matrix"]))
return input_values, sess.run(