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.cc514 Node* engine_node = graph->AddNode(trt_node, &status); in CreateTRTNode() local
515 (*engine_nodes)[pos] = engine_node; in CreateTRTNode()
523 << engine_node->name(); in CreateTRTNode()
524 graph->AddControlEdge(in, engine_node); in CreateTRTNode()
531 << " to " << engine_node->name() << ":" << i; in CreateTRTNode()
532 graph->AddEdge(n, in.index, engine_node, i); in CreateTRTNode()
548 VLOG(1) << "Updating control edge from " << engine_node->name() << " to " in CreateTRTNode()
551 graph->AddControlEdge(engine_node, output_node); in CreateTRTNode()
553 VLOG(1) << "Updating data edge from " << engine_node->name() << ":" in CreateTRTNode()
558 graph->UpdateEdge(engine_node, conn.port_number, output_node, port)); in CreateTRTNode()