提交 07406a0e 编写于 作者: K Karim Nosir 提交者: TensorFlower Gardener

Add generated tests for int8 type for mirror_pad

PiperOrigin-RevId: 306527646
Change-Id: I8d5582f12097ef94d0d9d8640f5a5b2cb48df3d3
上级 5394d6e9
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册