diff options
Diffstat (limited to 'tensorflow/core/framework/graph_to_functiondef.cc')
-rw-r--r-- | tensorflow/core/framework/graph_to_functiondef.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/framework/graph_to_functiondef.cc b/tensorflow/core/framework/graph_to_functiondef.cc index 4ffa503379..b2bc414c49 100644 --- a/tensorflow/core/framework/graph_to_functiondef.cc +++ b/tensorflow/core/framework/graph_to_functiondef.cc @@ -153,7 +153,7 @@ Status GraphToFunctionDef(const Graph& graph, const string& name, const string normalized = node_names.Normalize(node->name()); argdef->set_name(normalized); Edge const* edge; - TF_CHECK_OK(node->input_edge(0, &edge)); + TF_RETURN_IF_ERROR(node->input_edge(0, &edge)); return_values[normalized] = strings::StrCat(edge->src()->name(), ":", edge->src_output()); continue; |