Home
last modified time | relevance | path

Searched defs:execution_context (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc40 nvinfer1::IExecutionContext* execution_context = in Create() local
46 const nvinfer1::IExecutionContext* execution_context, in GetTrtBindingShape()
97 nvinfer1::IExecutionContext* execution_context, in SetTrtEngineInputs()
180 nvinfer1::IExecutionContext* execution_context, in SetTrtEngineOutputs()
243 Status TrtEnqueue(nvinfer1::IExecutionContext* execution_context, in TrtEnqueue()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc82 Status AllocateDeviceMemory(nvinfer1::IExecutionContext* execution_context, in AllocateDeviceMemory()
861 nvinfer1::IExecutionContext* execution_context; in ExecuteTrtEngine() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1411 TrtUniquePtrType<nvinfer1::IExecutionContext> execution_context( in BuildAndRun() local