Searched refs:prof_idx (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles_test.cc | 194 int prof_idx = in TEST_F() local
196 ASSERT_GE(prof_idx, 0); in TEST_F()
200 j, prof_idx, nvinfer1::OptProfileSelector::kMIN); in TEST_F()
202 j, prof_idx, nvinfer1::OptProfileSelector::kMAX); in TEST_F()
204 j, prof_idx, nvinfer1::OptProfileSelector::kOPT); in TEST_F()
|
| D | trt_shape_optimization_profiles.cc | 288 for (int prof_idx = 0; prof_idx < n_profiles; prof_idx++) { in RestoreProfiles() local
292 j, prof_idx, nvinfer1::OptProfileSelector::kMIN); in RestoreProfiles()
294 j, prof_idx, nvinfer1::OptProfileSelector::kMAX); in RestoreProfiles()
296 j, prof_idx, nvinfer1::OptProfileSelector::kOPT); in RestoreProfiles()
|