Searched refs:trt_algorithm_id (Results 1 – 1 of 1) sorted by relevance
1470 static int32_t trt_algorithm_id = [] { in BuildCudaEngine() local1471 int64 trt_algorithm_id; in BuildCudaEngine() local1474 &trt_algorithm_id)); in BuildCudaEngine()1475 return static_cast<int32_t>(trt_algorithm_id); in BuildCudaEngine()1478 if (trt_algorithm_id >= 0) { in BuildCudaEngine()1479 VLOG(1) << "Forcing TRT algorithm selection to: ID=" << trt_algorithm_id; in BuildCudaEngine()1480 StaticAlgorithmSelector trt_algorithm_selector(trt_algorithm_id); in BuildCudaEngine()