Home
last modified time | relevance | path

Searched refs:EngineInfo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc200 EngineInfo* info) { in GetEngineInfo()
359 void UpdateToEngineNode(const std::vector<EngineInfo>& infos, in UpdateToEngineNode()
401 const std::vector<EngineInfo>& infos, int pos, in CreateTRTNode()
463 if (info.engine_type == EngineInfo::EngineType::TRTStatic) { in CreateTRTNode()
506 if (info.engine_type == EngineInfo::EngineType::TRTStatic || calibrate_int8) { in CreateTRTNode()
549 if (info.engine_type == EngineInfo::EngineType::TRTStatic && in CreateTRTNode()
558 info.engine_type == EngineInfo::EngineType::TRTStatic) in CreateTRTNode()
731 const EngineInfo& engine) { in GetDeviceAndAllocator()
834 std::vector<EngineInfo> engine_segments; in ConvertAfterShapes()
844 EngineInfo curr_engine; in ConvertAfterShapes()
[all …]
Dconvert_graph.h97 const EngineInfo& engine);
Dconvert_nodes.h97 struct EngineInfo { struct
98 EngineInfo() in EngineInfo() function
Dconvert_graph_test.cc152 EngineInfo engine_info; in TEST()