Searched refs:EngineInfo (Results 1 – 4 of 4) sorted by relevance
200 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 …]
97 const EngineInfo& engine);
97 struct EngineInfo { struct98 EngineInfo() in EngineInfo() argument
152 EngineInfo engine_info; in TEST()