Searched defs:cuda_engine (Results 1 – 4 of 4) sorted by relevance
39 ExecutionContext ExecutionContext::Create(nvinfer1::ICudaEngine* cuda_engine) { in Create()45 Status GetTrtBindingShape(const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingShape()72 const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingIndex()96 Status SetTrtEngineInputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineInputs()179 Status SetTrtEngineOutputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineOutputs()
134 TrtUniquePtrType<nvinfer1::ICudaEngine> cuda_engine; member
450 int input_index, int binding_index, nvinfer1::ICudaEngine* cuda_engine, in SetInputShapeBinding()
834 auto& cuda_engine = engine_context->cuda_engine; in ExecuteTrtEngine() local