Home
last modified time | relevance | path

Searched refs:engine_node (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc510 Node* engine_node = graph->AddNode(trt_node, &status); in CreateTRTNode() local
511 (*engine_nodes)[pos] = engine_node; in CreateTRTNode()
519 << engine_node->name(); in CreateTRTNode()
520 graph->AddControlEdge(in, engine_node); in CreateTRTNode()
527 << " to " << engine_node->name() << ":" << i; in CreateTRTNode()
528 graph->AddEdge(n, in.index, engine_node, i); in CreateTRTNode()
544 VLOG(1) << "Updating control edge from " << engine_node->name() << " to " in CreateTRTNode()
547 graph->AddControlEdge(engine_node, output_node); in CreateTRTNode()
549 VLOG(1) << "Updating data edge from " << engine_node->name() << ":" in CreateTRTNode()
554 graph->UpdateEdge(engine_node, conn.port_number, output_node, port)); in CreateTRTNode()