Lines Matching refs:queryIdx
82 __device__ void loadQueryToSmem(int queryIdx, const PtrStepSz<T>& query, U* s_query) in loadQueryToSmem() argument
88 …eadIdx.y * MAX_DESC_LEN + loadX] = loadX < query.cols ? query.ptr(::min(queryIdx, query.rows - 1))… in loadQueryToSmem()
93 …__device__ void loopUnrolledCached(int queryIdx, const PtrStepSz<T>& query,volatile int imgIdx, co… in loopUnrolledCached() argument
129 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loopUnrolledCached()
143 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolledCached() local
148 loadQueryToSmem<BLOCK_SIZE, MAX_DESC_LEN>(queryIdx, query, s_query); in matchUnrolledCached()
153 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_tra… in matchUnrolledCached()
162 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolledCached()
164 bestTrainIdx[queryIdx] = myBestTrainIdx; in matchUnrolledCached()
165 bestDistance[queryIdx] = myBestDistance; in matchUnrolledCached()
192 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolledCached() local
197 loadQueryToSmem<BLOCK_SIZE, MAX_DESC_LEN>(queryIdx, query, s_query); in matchUnrolledCached()
209 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_t… in matchUnrolledCached()
220 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolledCached()
222 bestTrainIdx[queryIdx] = myBestTrainIdx; in matchUnrolledCached()
223 bestImgIdx[queryIdx] = myBestImgIdx; in matchUnrolledCached()
224 bestDistance[queryIdx] = myBestDistance; in matchUnrolledCached()
249 …__device__ void loopUnrolled(int queryIdx, const PtrStepSz<T>& query,volatile int imgIdx, const Pt… in loopUnrolled() argument
269 ForceGlob<T>::Load(query.ptr(::min(queryIdx, query.rows - 1)), loadX, val); in loopUnrolled()
289 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loopUnrolled()
303 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolled() local
311 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, my… in matchUnrolled()
320 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolled()
322 bestTrainIdx[queryIdx] = myBestTrainIdx; in matchUnrolled()
323 bestDistance[queryIdx] = myBestDistance; in matchUnrolled()
350 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolled() local
365 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, … in matchUnrolled()
376 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolled()
378 bestTrainIdx[queryIdx] = myBestTrainIdx; in matchUnrolled()
379 bestImgIdx[queryIdx] = myBestImgIdx; in matchUnrolled()
380 bestDistance[queryIdx] = myBestDistance; in matchUnrolled()
405 …__device__ void loop(int queryIdx, const PtrStepSz<T>& query, volatile int imgIdx, const PtrStepSz… in loop() argument
424 ForceGlob<T>::Load(query.ptr(::min(queryIdx, query.rows - 1)), loadX, val); in loop()
444 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loop()
458 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in match() local
466 …loop<BLOCK_SIZE, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, myBestDistance, myBestTr… in match()
475 if (queryIdx < query.rows && threadIdx.x == 0) in match()
477 bestTrainIdx[queryIdx] = myBestTrainIdx; in match()
478 bestDistance[queryIdx] = myBestDistance; in match()
505 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in match() local
519 …loop<BLOCK_SIZE, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, myBestDistance, myBest… in match()
530 if (queryIdx < query.rows && threadIdx.x == 0) in match()
532 bestTrainIdx[queryIdx] = myBestTrainIdx; in match()
533 bestImgIdx[queryIdx] = myBestImgIdx; in match()
534 bestDistance[queryIdx] = myBestDistance; in match()