TOCO: Print error message if conversion fails.

shashishekhar · tensorflower-gardener · commit 753cb67ab58f · 2019-02-10T11:35:12.000-08:00
PiperOrigin-RevId: 233301060
diff --git a/tensorflow/lite/toco/toco.cc b/tensorflow/lite/toco/toco.cc
@@ -49,5 +49,10 @@ int main(int argc, char** argv) {
   }
   toco::port::InitGoogle(argv[0], effective_argc, &effective_argv, true);
   auto status = toco::Convert(parsed_toco_flags, parsed_model_flags);
-  return status.ok() ? 0 : -1;
+  if (!status.ok()) {
+    fprintf(stderr, "%s\n", status.error_message().c_str());
+    fflush(stderr);
+    return 1;
+  }
+  return 0;
 }

Original file line number	Diff line number	Diff line change
`@@ -49,5 +49,10 @@ int main(int argc, char** argv) {`
`49`	`49`	`}`
`50`	`50`	`toco::port::InitGoogle(argv[0], effective_argc, &effective_argv, true);`
`51`	`51`	`auto status = toco::Convert(parsed_toco_flags, parsed_model_flags);`
`52`		`- return status.ok() ? 0 : -1;`
	`52`	`+ if (!status.ok()) {`
	`53`	`+ fprintf(stderr, "%s\n", status.error_message().c_str());`
	`54`	`+ fflush(stderr);`
	`55`	`+ return 1;`
	`56`	`+ }`
	`57`	`+ return 0;`
`53`	`58`	`}`