Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.h134 Status SetTrtEngineInputs(nvinfer1::ICudaEngine* cuda_engine,
Dtrt_engine_utils.cc134 Status SetTrtEngineInputs(nvinfer1::ICudaEngine* cuda_engine, in SetTrtEngineInputs() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc784 TF_RETURN_IF_ERROR(SetTrtEngineInputs(cuda_engine.get(), execution_context, in ExecuteTrtEngine()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1514 TF_RETURN_IF_ERROR(SetTrtEngineInputs( in BuildAndRun()