Home
last modified time | relevance | path

Searched refs:GetNumberOfEngineInputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h182 int GetNumberOfEngineInputs(const nvinfer1::ICudaEngine* engine);
Dutils.cc241 int GetNumberOfEngineInputs(const nvinfer1::ICudaEngine* engine) { in GetNumberOfEngineInputs() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc285 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles()