Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc283 int n_profiles = engine->getNbOptimizationProfiles(); in RestoreProfiles() local
284 need_profiles_ = n_profiles > 0; in RestoreProfiles()
286 VLOG(2) << "Attempting to restore " << n_profiles << " profiles, each with " in RestoreProfiles()
288 for (int prof_idx = 0; prof_idx < n_profiles; prof_idx++) { in RestoreProfiles()
Dtrt_engine_utils.cc121 int n_profiles = cuda_engine->getNbOptimizationProfiles(); in GetTrtBindingIndex() local
123 int n_profiles = 1; in GetTrtBindingIndex() local
129 const int bindings_per_profile = cuda_engine->getNbBindings() / n_profiles; in GetTrtBindingIndex()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc253 int n_profiles = engine->getNbOptimizationProfiles(); in GetNumberOfEngineInputs() local
255 int n_profiles = 1; in GetNumberOfEngineInputs() local
257 return n_input / n_profiles; in GetNumberOfEngineInputs()