Searched refs:engine_nodes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 361 const std::vector<Node*>& engine_nodes, in UpdateToEngineNode() argument
376 *node = CHECK_NOTNULL(engine_nodes[t]); in UpdateToEngineNode()
404 std::vector<Node*>* engine_nodes) { in CreateTRTNode() argument
428 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
478 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
582 (*engine_nodes)[pos] = engine_node; in CreateTRTNode()
611 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/false, in CreateTRTNode()
894 std::vector<Node*> engine_nodes; in ConvertAfterShapes() local
895 engine_nodes.resize(engine_segments.size()); in ConvertAfterShapes()
921 alloc.get(), &engine_nodes); in ConvertAfterShapes()
|