Searched defs:cuda_engine (Results 1 – 3 of 3) sorted by relevance
50 nvinfer1::ICudaEngine* cuda_engine, TRTBaseAllocator* allocator) { in Create()74 Status GetTrtBindingShape(const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingShape()106 const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingIndex()134 Status SetTrtEngineInputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineInputs()217 Status SetTrtEngineOutputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineOutputs()
134 TrtUniquePtrType<nvinfer1::ICudaEngine> cuda_engine; member
748 auto& cuda_engine = engine_context->cuda_engine; in ExecuteTrtEngine() local