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

Addressed review comments.

Browse Source 
Change-Id: I12b1bfc081dd04f4ab5480ea8bafe280f0e588e5
This commit is contained in:
Elena Zhelezina 2020-07-16 11:38:59 +01:00
parent 261d74db28
commit 374a1e7a63
1 changed files with 3 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tensorflow/lite/testing/op_tests/pad.py
View File

@ -67,17 +67,7 @@ def make_pad_tests(options):
[[0, 0], [0, 0], [0, 0], [0, 0]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [False]
},
{
"dtype": [tf.float32],
"input_shape": [[1, 1, 2, 1], [2, 1, 1, 1]],
"paddings": [[[0, 0], [0, 1], [2, 3], [0, 0]],
[[0, 1], [0, 0], [0, 0], [2, 3]],
[[0, 0], [0, 0], [0, 0], [0, 0]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [True]
"quant_16x8": [False, True]
},
# 2D:
{
@ -86,15 +76,7 @@ def make_pad_tests(options):
"paddings": [[[0, 1], [2, 3]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [False],
},
{
"dtype": [tf.float32],
"input_shape": [[1, 2]],
"paddings": [[[0, 1], [2, 3]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [True],
"quant_16x8": [False, True],
},
# 1D:
{
@ -103,15 +85,7 @@ def make_pad_tests(options):
"paddings": [[[1, 2]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [False],
},
{
"dtype": [tf.float32],
"input_shape": [[1]],
"paddings": [[[1, 2]]],
"constant_paddings": [True],
"fully_quantize": [True],
"quant_16x8": [True],
"quant_16x8": [False, True],
},
]