diff options
author | 2017-12-22 12:21:16 -0800 | |
---|---|---|
committer | 2017-12-22 12:25:02 -0800 | |
commit | 4ff57df76b9013cf22550aac8fc66717a2d0891d (patch) | |
tree | aedfc59fe4d6afcbbc7c2868a872df7156ac857b | |
parent | ba32ea1547af74d549f35a42e4de83c88652a636 (diff) |
Uses ImportGraphDef instead of ConvertGraphDefToGraph in tflite.
This has the right semantics for backwards compatibility.
PiperOrigin-RevId: 179951639
-rw-r--r-- | tensorflow/contrib/lite/toco/import_tensorflow.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.cc b/tensorflow/contrib/lite/toco/import_tensorflow.cc index d56f72d44f..866898cc3d 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.cc +++ b/tensorflow/contrib/lite/toco/import_tensorflow.cc @@ -1709,11 +1709,11 @@ bool InlineAllFunctions(GraphDef* graphdef) { flr = pflr.GetFLR("/job:localhost/replica:0/task:0/cpu:0"); tensorflow::Graph graph(fld); - tensorflow::GraphConstructorOptions gc_opts; - const auto& tf_convert_status = - tensorflow::ConvertGraphDefToGraph(gc_opts, graphdef_copy, &graph); + tensorflow::ImportGraphDefOptions gc_opts; + const auto& tf_convert_status = tensorflow::ImportGraphDef( + gc_opts, graphdef_copy, &graph, nullptr, nullptr); if (!tf_convert_status.ok()) { - LOG(ERROR) << "tensorflow::ConvertGraphDefToGraph failed with status: " + LOG(ERROR) << "tensorflow::ImportGraphDef failed with status: " << tf_convert_status.ToString(); return false; } |