Searched refs:engine_nodes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 292 const std::vector<Node*>& engine_nodes, in UpdateToEngineNode() argument
307 *node = CHECK_NOTNULL(engine_nodes[t]); in UpdateToEngineNode()
334 std::vector<Node*>* engine_nodes) { in CreateTRTNode() argument
357 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
400 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
511 (*engine_nodes)[pos] = engine_node; in CreateTRTNode()
540 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/false, in CreateTRTNode()
830 std::vector<Node*> engine_nodes; in ConvertAfterShapes() local
831 engine_nodes.resize(engine_segments.size()); in ConvertAfterShapes()
840 params.max_batch_size, &graph, &engine_nodes); in ConvertAfterShapes()
|