Home
last modified time | relevance | path

Searched refs:TrtEnqueue (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.h163 Status TrtEnqueue(nvinfer1::IExecutionContext* execution_context,
Dtrt_engine_utils.cc279 Status TrtEnqueue(nvinfer1::IExecutionContext* execution_context, in TrtEnqueue() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc799 TF_RETURN_IF_ERROR(TrtEnqueue(execution_context, buffers, *stream, in ExecuteTrtEngine()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1522 TF_RETURN_IF_ERROR(TrtEnqueue(execution_context.get(), buffers, stream_, in BuildAndRun()