Searched refs:cuda_engine (Results 1 – 2 of 2) sorted by relevance
378 if (!engine_context->cuda_engine) { in ComputeAsync()397 auto& cuda_engine = engine_context->cuda_engine; in ExecuteTrtEngine() local406 const int binding_index = cuda_engine->getBindingIndex(input_name.c_str()); in ExecuteTrtEngine()422 auto dtype = cuda_engine->getBindingDataType(binding_index); in ExecuteTrtEngine()447 const int binding_index = cuda_engine->getBindingIndex(output_name.c_str()); in ExecuteTrtEngine()452 auto dims = cuda_engine->getBindingDimensions(binding_index); in ExecuteTrtEngine()477 auto dtype = cuda_engine->getBindingDataType(binding_index); in ExecuteTrtEngine()
133 : cuda_engine(std::move(input_cuda_engine)), in EngineContext()137 TrtUniquePtrType<nvinfer1::ICudaEngine> cuda_engine; member168 << "ICudaEngine: " << item.second.get()->cuda_engine.get() << ", " in DebugString()