Searched refs:builder_config (Results 1 – 1 of 1) sorted by relevance
1465 TrtUniquePtrType<nvinfer1::IBuilderConfig> builder_config( in BuildCudaEngine() local1467 builder_config->setMaxWorkspaceSize(max_workspace_size_bytes); in BuildCudaEngine()1481 builder_config->setAlgorithmSelector(&trt_algorithm_selector); in BuildCudaEngine()1486 builder_config->setFlag(nvinfer1::BuilderFlag::kFP16); in BuildCudaEngine()1488 builder_config->setFlag(nvinfer1::BuilderFlag::kFP16); in BuildCudaEngine()1489 builder_config->setFlag(nvinfer1::BuilderFlag::kINT8); in BuildCudaEngine()1491 builder_config->setInt8Calibrator(calibrator); in BuildCudaEngine()1493 builder_config->setInt8Calibrator(nullptr); in BuildCudaEngine()1498 trt_builder_.get(), builder_config.get(), network())); in BuildCudaEngine()1515 trt_builder_->buildEngineWithConfig(*network(), *builder_config)); in BuildCudaEngine()