Searched refs:trt_builder_ (Results 1 – 2 of 2) sorted by relevance
1267 trt_builder_.reset(nvinfer1::createInferBuilder(*trt_logger)); in Init()1276 trt_network_.reset(trt_builder_->createNetworkV2(flags)); in Init()1278 trt_network_.reset(trt_builder_->createNetwork()); in Init()1461 trt_builder_->setMaxBatchSize(max_batch_size); in BuildCudaEngine()1462 trt_builder_->setGpuAllocator(allocator); in BuildCudaEngine()1466 trt_builder_->createBuilderConfig()); in BuildCudaEngine()1498 trt_builder_.get(), builder_config.get(), network())); in BuildCudaEngine()1515 trt_builder_->buildEngineWithConfig(*network(), *builder_config)); in BuildCudaEngine()1517 trt_builder_->setMaxWorkspaceSize(max_workspace_size_bytes); in BuildCudaEngine()1519 trt_builder_->setFp16Mode(true); in BuildCudaEngine()[all …]
641 TrtUniquePtrType<nvinfer1::IBuilder> trt_builder_; variable