From 50ba36f1662dc61cb1b60353a2a09aa3ea72bb59 Mon Sep 17 00:00:00 2001 From: Jared Duke Date: Mon, 30 Jul 2018 10:53:43 -0700 Subject: Fix issue with scalar reshape parameters PiperOrigin-RevId: 206609783 --- tensorflow/contrib/lite/testing/generate_examples.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'tensorflow/contrib/lite/testing') 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): -- cgit v1.2.3