Searched defs:cuda_engine (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| H A D | trt_engine_utils.cc | 39 ExecutionContext ExecutionContext::Create(nvinfer1::ICudaEngine* cuda_engine) { in Create()
60 Status GetTrtBindingShape(const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingShape()
83 Status SetupBindings(nvinfer1::ICudaEngine* cuda_engine, const Tensor& tensor, in SetupBindings()
115 Status SetTrtEngineInputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineInputs()
204 Status SetTrtEngineOutputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineOutputs()
|
| H A D | trt_shape_optimization_profiles.cc | 519 int input_index, int binding_index, nvinfer1::ICudaEngine* cuda_engine, in SetInputShapeBinding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| H A D | utils.cc | 61 const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingIndex()
87 const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingIndex()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 948 nvinfer1::ICudaEngine* cuda_engine = engine_context->GetCudaEngine(); in ExecuteTrtEngine() local
|