Searched refs:UpdateToEngineNode (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 359 void UpdateToEngineNode(const std::vector<EngineInfo>& infos, in UpdateToEngineNode() function
428 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
478 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/true, in CreateTRTNode()
611 UpdateToEngineNode(infos, pos, *engine_nodes, /*is_input_edge=*/false, in CreateTRTNode()
|