diff options
author | 2018-08-09 17:47:31 -0700 | |
---|---|---|
committer | 2018-08-09 17:51:23 -0700 | |
commit | c36b51458563e9f72c0d37ddf84206f639e13f72 (patch) | |
tree | d46bea1cdc384e35d257dd6606a97d4783b999b1 /tensorflow/contrib/lite/python | |
parent | 5a049ffcb72e09305e08631ccb7a68236a61f825 (diff) |
Update code to use new flag name: dump_graphviz_dir
PiperOrigin-RevId: 208141063
Diffstat (limited to 'tensorflow/contrib/lite/python')
-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 d17482e601..a76cc39635 100644 --- a/tensorflow/contrib/lite/python/tflite_convert.py +++ b/tensorflow/contrib/lite/python/tflite_convert.py @@ -203,8 +203,9 @@ def _check_flags(flags, unparsed): raise ValueError("--default_ranges_min and --default_ranges_max must be " "used together") - if flags.dump_graphviz_video and not flags.dump_graphviz: - raise ValueError("--dump_graphviz_video must be used with --dump_graphviz") + if flags.dump_graphviz_video and not flags.dump_graphviz_dir: + raise ValueError("--dump_graphviz_video must be used with " + "--dump_graphviz_dir") def run_main(_): |