diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/flatten_call_graph.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/flatten_call_graph.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/flatten_call_graph.cc b/tensorflow/compiler/xla/service/flatten_call_graph.cc index 3c41fe870f..297a4f7599 100644 --- a/tensorflow/compiler/xla/service/flatten_call_graph.cc +++ b/tensorflow/compiler/xla/service/flatten_call_graph.cc @@ -102,8 +102,7 @@ Status FlattenNode(const CallGraphNode& node) { StatusOr<bool> FlattenCallGraph::Run(HloModule* module) { XLA_VLOG_LINES(3, "Before flatten call graph:\n" + module->ToString()); - TF_ASSIGN_OR_RETURN(std::unique_ptr<CallGraph> call_graph, - CallGraph::Build(module)); + std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); TF_RETURN_IF_ERROR(call_graph->VisitNodes(FlattenNode)); XLA_VLOG_LINES(3, "After flatten call graph:\n" + module->ToString()); |