diff options
| author |  Suharsh Sivakumar <suharshs@google.com> | 2018-06-11 18:41:48 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-11 18:44:42 -0700 |
| commit | b5fa781337ad8becaab893d001b04f2b995575b5 (patch) | |
| tree | 2f96ca84c22f832617eb8b71224324f9b96463da /tensorflow/contrib/lite/python/tflite_convert.py | |
| parent | 8c5d37c3b96cdbcb8a3b657144d4fb63fb3dc100 (diff) | |
TFLite should allow values of 0 for default_ranges_{min,max}.
PiperOrigin-RevId: 200149066
Diffstat (limited to 'tensorflow/contrib/lite/python/tflite_convert.py')
| -rw-r--r-- | tensorflow/contrib/lite/python/tflite_convert.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/python/tflite_convert.py b/tensorflow/contrib/lite/python/tflite_convert.py index 32ad84ec3c..f497533bed 100644 --- a/tensorflow/contrib/lite/python/tflite_convert.py +++ b/tensorflow/contrib/lite/python/tflite_convert.py @@ -116,7 +116,8 @@ def _convert_model(flags): "tensors in order to map between names and " "values.".format(",".join(input_arrays))) converter.quantized_input_stats = dict(zip(input_arrays, quant_stats)) - if flags.default_ranges_min and flags.default_ranges_max: + if (flags.default_ranges_min is not None) and (flags.default_ranges_max is + not None): converter.default_ranges_stats = (flags.default_ranges_min, flags.default_ranges_max) @@ -195,7 +196,7 @@ def _check_flags(flags, unparsed): raise ValueError("--std_dev_values, --mean_values must have the same " "number of items") - if bool(flags.default_ranges_min) != bool(flags.default_ranges_max): + if (flags.default_ranges_min is None) != (flags.default_ranges_max is None): raise ValueError("--default_ranges_min and --default_ranges_max must be " "used together") |