Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc121 int n_profiles = cuda_engine->getNbOptimizationProfiles(); in GetTrtBindingIndex() local
123 int n_profiles = 1; in GetTrtBindingIndex() local
Dtrt_shape_optimization_profiles.cc283 int n_profiles = engine->getNbOptimizationProfiles(); in RestoreProfiles() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc253 int n_profiles = engine->getNbOptimizationProfiles(); in GetNumberOfEngineInputs() local
255 int n_profiles = 1; in GetNumberOfEngineInputs() local