diff options
author | 2018-07-30 10:53:43 -0700 | |
---|---|---|
committer | 2018-07-30 10:57:45 -0700 | |
commit | 50ba36f1662dc61cb1b60353a2a09aa3ea72bb59 (patch) | |
tree | 61383a4a9792353e7699fb6d995ac724acc7026d /tensorflow/contrib/lite/testing | |
parent | 538c7198fdba4ac9b71ba2ceb8c2fb0bb31d20e5 (diff) |
Fix issue with scalar reshape parameters
PiperOrigin-RevId: 206609783
Diffstat (limited to 'tensorflow/contrib/lite/testing')
-rw-r--r-- | tensorflow/contrib/lite/testing/generate_examples.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/generate_examples.py b/tensorflow/contrib/lite/testing/generate_examples.py index a95b26220d..6d03c0fd9e 100644 --- a/tensorflow/contrib/lite/testing/generate_examples.py +++ b/tensorflow/contrib/lite/testing/generate_examples.py @@ -1615,6 +1615,11 @@ def make_reshape_tests(zip_path): "input_shape": [[3, 4, 5, 7], [4, 105], [21, 5, 2, 2], [420]], "output_shape": [[15, 28], [420], [1, -1, 5, 7], [-1]], "constant_shape": [True, False], + }, { + "dtype": [tf.float32], + "input_shape": [[1]], + "output_shape": [[]], + "constant_shape": [True, False], }] def build_graph(parameters): |