Searched refs:sgraph (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 634 Graph sgraph(graph->flib_def()); in RegisterSegmentFunctionToFunctionLibrary() local
636 TF_RETURN_IF_ERROR(ConvertGraphDefToGraph(gcopts, segment, &sgraph)); in RegisterSegmentFunctionToFunctionLibrary()
639 for (auto n : sgraph.op_nodes()) { in RegisterSegmentFunctionToFunctionLibrary()
659 auto node_arg = sgraph.AddNode(nd, &s); in RegisterSegmentFunctionToFunctionLibrary()
664 sgraph.AddEdge(node_arg, 0, edge->dst(), edge->dst_input()); in RegisterSegmentFunctionToFunctionLibrary()
672 sgraph.RemoveNode(node); in RegisterSegmentFunctionToFunctionLibrary()
696 auto node_ret = sgraph.AddNode(nd, &s); in RegisterSegmentFunctionToFunctionLibrary()
702 sgraph.AddEdge(edge->src(), edge->src_output(), node_ret, 0); in RegisterSegmentFunctionToFunctionLibrary()
703 s = sgraph.UpdateEdge(edge->src(), edge->src_output(), node_ret, 0); in RegisterSegmentFunctionToFunctionLibrary()
709 sgraph.RemoveNode(node); in RegisterSegmentFunctionToFunctionLibrary()
[all …]
|