Home
last modified time | relevance | path

Searched refs:GetTrtBindingShape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc74 Status GetTrtBindingShape(const nvinfer1::ICudaEngine* cuda_engine, in GetTrtBindingShape() function
233 TF_RETURN_IF_ERROR(GetTrtBindingShape(cuda_engine, execution_context, in SetTrtEngineOutputs()
Dtrt_engine_utils.h146 Status GetTrtBindingShape(const nvinfer1::ICudaEngine* cuda_engine,