Searched refs:gpu_indices (Results 1 – 4 of 4) sorted by relevance
126 std::vector<int> gpu_indices = CSVToVector<int>(sift_options_.gpu_index); in SiftFeatureExtractor() local127 CHECK_GT(gpu_indices.size(), 0); in SiftFeatureExtractor()130 if (gpu_indices.size() == 1 && gpu_indices[0] == -1) { in SiftFeatureExtractor()133 gpu_indices.resize(num_cuda_devices); in SiftFeatureExtractor()134 std::iota(gpu_indices.begin(), gpu_indices.end(), 0); in SiftFeatureExtractor()139 for (const auto& gpu_index : gpu_indices) { in SiftFeatureExtractor()
770 if (!options.darkness_adaptivity && gpu_indices[0] < 0) { in CreateSiftGPUExtractor()771 gpu_indices[0] = 0; in CreateSiftGPUExtractor()774 if (gpu_indices[0] >= 0) { in CreateSiftGPUExtractor()783 if (gpu_indices[0] >= 0) { in CreateSiftGPUExtractor()837 sift_gpu->gpu_index = gpu_indices[0]; in CreateSiftGPUExtractor()838 if (sift_extraction_mutexes.count(gpu_indices[0]) == 0) { in CreateSiftGPUExtractor()1094 const std::vector<int> gpu_indices = in CreateSiftGPUMatcher() local1104 if (gpu_indices[0] >= 0) { in CreateSiftGPUMatcher()1106 gpu_indices[0]); in CreateSiftGPUMatcher()1139 sift_match_gpu->gpu_index = gpu_indices[0]; in CreateSiftGPUMatcher()[all …]
616 std::vector<int> gpu_indices = CSVToVector<int>(options_.gpu_index); in SiftFeatureMatcher() local617 CHECK_GT(gpu_indices.size(), 0); in SiftFeatureMatcher()620 if (options_.use_gpu && gpu_indices.size() == 1 && gpu_indices[0] == -1) { in SiftFeatureMatcher()623 gpu_indices.resize(num_cuda_devices); in SiftFeatureMatcher()624 std::iota(gpu_indices.begin(), gpu_indices.end(), 0); in SiftFeatureMatcher()630 matchers_.reserve(gpu_indices.size()); in SiftFeatureMatcher()631 for (const auto& gpu_index : gpu_indices) { in SiftFeatureMatcher()653 guided_matchers_.reserve(gpu_indices.size()); in SiftFeatureMatcher()654 for (const auto& gpu_index : gpu_indices) { in SiftFeatureMatcher()
99 const std::vector<int> gpu_indices = CSVToVector<int>(options_.gpu_index); in Check() local100 CHECK_EQ(gpu_indices.size(), 1); in Check()101 CHECK_GE(gpu_indices[0], -1); in Check()191 std::vector<int> gpu_indices = CSVToVector<int>(options_.gpu_index); in PatchMatchController() local