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

add bias to transpose_conv TESTs.

Browse Source
This commit is contained in:
Peng Sun 2020-05-11 08:53:29 +01:00
parent 0ed61bf948
commit b76ef65778
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tensorflow/lite/testing/op_tests/transpose_conv.py
View File

@ -38,6 +38,7 @@ def make_transpose_conv_tests(options):
{
"input_shape": [[1, 3, 4, 1], [1, 10, 10, 3], [3, 20, 20, 1]],
"filter_size": [[1, 1], [1, 2], [3, 3]],
"has_bias": [False],
"strides": [[1, 1, 1, 1], [1, 3, 3, 1]],
"padding": ["SAME", "VALID"],
"data_format": ["NHWC"],
@ -50,6 +51,7 @@ def make_transpose_conv_tests(options):
{
"input_shape": [[1, 3, 3, 1]],
"filter_size": [[3, 3, 2, 1]],
"has_bias": [False],
"strides": [[1, 1, 1, 1]],
"padding": ["SAME"],
"data_format": ["NHWC"],
@ -60,6 +62,7 @@ def make_transpose_conv_tests(options):
{
"input_shape": [[1, 3, 3, 1]],
"filter_size": [[3, 3, 2, 1]],
"has_bias": [False],
"strides": [[1, 2, 2, 1]],
"padding": ["SAME"],
"data_format": ["NHWC"],
@ -70,13 +73,26 @@ def make_transpose_conv_tests(options):
{
"input_shape": [[1, 4, 3, 1]],
"filter_size": [[3, 3, 2, 1]],
"has_bias": [False],
"strides": [[1, 2, 2, 1]],
"padding": ["SAME"],
"data_format": ["NHWC"],
"channel_multiplier": [1],
"output_shape": [[1, 8, 6, 2]],
"fully_quantize": [True]
}
},
{
"input_shape": [[1, 3, 3, 1]],
"filter_size": [[3, 3, 2, 1]],
"has_bias": [True],
"strides": [[1, 1, 1, 1]],
"padding": ["SAME"],
"data_format": ["NHWC"],
"channel_multiplier": [1],
"output_shape": [[1, 3, 3, 2]],
"fully_quantize": [True]
},
]
def get_tensor_shapes(parameters):
@ -124,6 +140,13 @@ def make_transpose_conv_tests(options):
strides=parameters["strides"],
padding=parameters["padding"],
data_format=parameters["data_format"])
if parameters["has_bias"]:
bias_input = create_tensor_data(np.float32,
(parameters["output_shape"][-1],),
min_value=-1,
max_value=1)
out = tf.nn.bias_add(out, bias_input,
data_format=parameters["data_format"])
return input_tensors, [out]