From c36b51458563e9f72c0d37ddf84206f639e13f72 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 9 Aug 2018 17:47:31 -0700 Subject: Update code to use new flag name: dump_graphviz_dir PiperOrigin-RevId: 208141063 --- tensorflow/contrib/lite/python/tflite_convert.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tensorflow/contrib/lite/python') 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(_): -- cgit v1.2.3