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