Lines Matching refs:queryIdx
305 __device__ void loadQueryToSmem(int queryIdx, const PtrStepSz<T>& query, U* s_query) in loadQueryToSmem() argument
311 …eadIdx.y * MAX_DESC_LEN + loadX] = loadX < query.cols ? query.ptr(::min(queryIdx, query.rows - 1))… in loadQueryToSmem()
316 …__device__ void loopUnrolledCached(int queryIdx, const PtrStepSz<T>& query, int imgIdx, const PtrS… in loopUnrolledCached() argument
354 if (queryIdx < query.rows && trainIdx < train.rows && mask(queryIdx, trainIdx)) in loopUnrolledCached()
381 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolledCached() local
386 loadQueryToSmem<BLOCK_SIZE, MAX_DESC_LEN>(queryIdx, query, s_query); in matchUnrolledCached()
393 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_tra… in matchUnrolledCached()
402 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolledCached()
404 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in matchUnrolledCached()
405 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in matchUnrolledCached()
431 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolledCached() local
436 loadQueryToSmem<BLOCK_SIZE, MAX_DESC_LEN>(queryIdx, query, s_query); in matchUnrolledCached()
451 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_t… in matchUnrolledCached()
462 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolledCached()
464 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in matchUnrolledCached()
465 bestImgIdx[queryIdx] = make_int2(myBestImgIdx1, myBestImgIdx2); in matchUnrolledCached()
466 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in matchUnrolledCached()
491 …__device__ void loopUnrolled(int queryIdx, const PtrStepSz<T>& query, int imgIdx, const PtrStepSz<… in loopUnrolled() argument
513 ForceGlob<T>::Load(query.ptr(::min(queryIdx, query.rows - 1)), loadX, val); in loopUnrolled()
533 if (queryIdx < query.rows && trainIdx < train.rows && mask(queryIdx, trainIdx)) in loopUnrolled()
560 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolled() local
570 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, my… in matchUnrolled()
579 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolled()
581 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in matchUnrolled()
582 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in matchUnrolled()
608 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in matchUnrolled() local
626 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, … in matchUnrolled()
637 if (queryIdx < query.rows && threadIdx.x == 0) in matchUnrolled()
639 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in matchUnrolled()
640 bestImgIdx[queryIdx] = make_int2(myBestImgIdx1, myBestImgIdx2); in matchUnrolled()
641 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in matchUnrolled()
666 …__device__ void loop(int queryIdx, const PtrStepSz<T>& query, int imgIdx, const PtrStepSz<T>& trai… in loop() argument
687 ForceGlob<T>::Load(query.ptr(::min(queryIdx, query.rows - 1)), loadX, val); in loop()
707 if (queryIdx < query.rows && trainIdx < train.rows && mask(queryIdx, trainIdx)) in loop()
734 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in match() local
744 …loop<BLOCK_SIZE, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, myBestDistance1, myBestD… in match()
753 if (queryIdx < query.rows && threadIdx.x == 0) in match()
755 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in match()
756 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in match()
782 const int queryIdx = blockIdx.x * BLOCK_SIZE + threadIdx.y; in match() local
800 …loop<BLOCK_SIZE, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, myBestDistance1, myBes… in match()
811 if (queryIdx < query.rows && threadIdx.x == 0) in match()
813 bestTrainIdx[queryIdx] = make_int2(myBestTrainIdx1, myBestTrainIdx2); in match()
814 bestImgIdx[queryIdx] = make_int2(myBestImgIdx1, myBestImgIdx2); in match()
815 bestDistance[queryIdx] = make_float2(myBestDistance1, myBestDistance2); in match()
909 const int queryIdx = blockIdx.y * BLOCK_SIZE + threadIdx.y; in calcDistanceUnrolled() local
924 …s_query[threadIdx.y * BLOCK_SIZE + threadIdx.x] = query.ptr(::min(queryIdx, query.rows - 1))[loadX… in calcDistanceUnrolled()
942 if (queryIdx < query.rows && trainIdx < train.rows) in calcDistanceUnrolled()
946 if (mask(queryIdx, trainIdx)) in calcDistanceUnrolled()
949 allDist.ptr(queryIdx)[trainIdx] = distVal; in calcDistanceUnrolled() local
973 const int queryIdx = blockIdx.y * BLOCK_SIZE + threadIdx.y; in calcDistance() local
987 …s_query[threadIdx.y * BLOCK_SIZE + threadIdx.x] = query.ptr(::min(queryIdx, query.rows - 1))[loadX… in calcDistance()
1005 if (queryIdx < query.rows && trainIdx < train.rows) in calcDistance()
1009 if (mask(queryIdx, trainIdx)) in calcDistance()
1012 allDist.ptr(queryIdx)[trainIdx] = distVal; in calcDistance() local
1075 const int queryIdx = blockIdx.x; in findBestMatch() local
1077 float* allDistRow = allDist.ptr(queryIdx); in findBestMatch()
1103 trainIdx.ptr(queryIdx)[i] = bestIdx; in findBestMatch() local
1104 distance.ptr(queryIdx)[i] = dist; in findBestMatch() local