diff options
author | Yunlu Li <yunluli@google.com> | 2018-08-30 17:19:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 17:24:24 -0700 |
commit | 9efc469cc5fa7eb0f8d8d8a7662cb6c1fbcb5b1a (patch) | |
tree | c01018a17eeb3c9397f4606c445e666af2d080e4 /tensorflow/contrib/lite/python/lite_test.py | |
parent | 11f970c61bc21ae81b76cdee58871f0509ec9f0f (diff) |
Change the data type of mean_values and std_dev_values to float.
PiperOrigin-RevId: 211010293
Diffstat (limited to 'tensorflow/contrib/lite/python/lite_test.py')
-rw-r--r-- | tensorflow/contrib/lite/python/lite_test.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/python/lite_test.py b/tensorflow/contrib/lite/python/lite_test.py index 8c9cfa943f..e6aa8b0d99 100644 --- a/tensorflow/contrib/lite/python/lite_test.py +++ b/tensorflow/contrib/lite/python/lite_test.py @@ -319,6 +319,7 @@ class FromSessionTest(test_util.TensorFlowTestCase): # Convert model and ensure model is not None. converter = lite.TocoConverter.from_session(sess, [in_tensor], [out_tensor]) converter.inference_input_type = lite_constants.QUANTIZED_UINT8 + converter.quantized_input_stats = {'Placeholder': (0., 1.)} # mean, std_dev tflite_model = converter.convert() self.assertTrue(tflite_model) |