Searched refs:engine_name (Results 1 – 5 of 5) sorted by relevance
338 info->engine_name = StrCat(scope_name, info->engine_name); in GetEngineInfo()339 VLOG(1) << "Converted TensorRT candidate segment '" << info->engine_name in GetEngineInfo()377 QCHECK_EQ(info.engine_name, (**node).name()) in UpdateToEngineNode()378 << "Engine name mismatch: " << info.engine_name << " vs " in UpdateToEngineNode()415 VLOG(1) << "Processing " << info.engine_name; in CreateTRTNode()437 << info.engine_name; in CreateTRTNode()489 << " -> " << info.engine_name << ":" << inputs.size() - 1; in CreateTRTNode()535 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode()538 string ins = StrCat(info.engine_name, " inputs= "); in CreateTRTNode()561 ? StrCat(info.engine_name, "_native_segment") in CreateTRTNode()[all …]
104 string engine_name; member
35 int batch_size, string engine_name) in TRTInt8Calibrator() argument42 engine_name_(engine_name) {} in TRTInt8Calibrator()
47 int batch_size, string engine_name);
371 for engine_name, node_names in expected_engines.items():373 old_to_new_node_map[node_name] = engine_name