Lines Matching refs:train
93 …eryIdx, const PtrStepSz<T>& query,volatile int imgIdx, const PtrStepSz<T>& train, const Mask& mask, in loopUnrolledCached() argument
97 for (int t = 0, endt = (train.rows + BLOCK_SIZE - 1) / BLOCK_SIZE; t < endt; ++t) in loopUnrolledCached()
108 if (loadX < train.cols) in loopUnrolledCached()
112 … ForceGlob<T>::Load(train.ptr(::min(t * BLOCK_SIZE + threadIdx.y, train.rows - 1)), loadX, val); in loopUnrolledCached()
129 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loopUnrolledCached()
139 …__global__ void matchUnrolledCached(const PtrStepSz<T> query, const PtrStepSz<T> train, const Mask… in matchUnrolledCached() argument
153 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_tra… in matchUnrolledCached()
170 … void matchUnrolledCached(const PtrStepSz<T>& query, const PtrStepSz<T>& train, const Mask& mask, in matchUnrolledCached() argument
179 …BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, trainIdx.da… in matchUnrolledCached()
207 const PtrStepSz<T> train = trains[imgIdx]; in matchUnrolledCached() local
209 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_t… in matchUnrolledCached()
249 …eryIdx, const PtrStepSz<T>& query,volatile int imgIdx, const PtrStepSz<T>& train, const Mask& mask, in loopUnrolled() argument
253 for (int t = 0, endt = (train.rows + BLOCK_SIZE - 1) / BLOCK_SIZE; t < endt; ++t) in loopUnrolled()
272 … ForceGlob<T>::Load(train.ptr(::min(t * BLOCK_SIZE + threadIdx.y, train.rows - 1)), loadX, val); in loopUnrolled()
289 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loopUnrolled()
299 …__global__ void matchUnrolled(const PtrStepSz<T> query, const PtrStepSz<T> train, const Mask mask,… in matchUnrolled() argument
311 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, my… in matchUnrolled()
328 void matchUnrolled(const PtrStepSz<T>& query, const PtrStepSz<T>& train, const Mask& mask, in matchUnrolled() argument
337 …BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, trainIdx.da… in matchUnrolled()
363 const PtrStepSz<T> train = trains[imgIdx]; in matchUnrolled() local
365 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, … in matchUnrolled()
405 …ryIdx, const PtrStepSz<T>& query, volatile int imgIdx, const PtrStepSz<T>& train, const Mask& mask, in loop() argument
409 for (int t = 0, endt = (train.rows + BLOCK_SIZE - 1) / BLOCK_SIZE; t < endt; ++t) in loop()
427 … ForceGlob<T>::Load(train.ptr(::min(t * BLOCK_SIZE + threadIdx.y, train.rows - 1)), loadX, val); in loop()
444 …if (queryIdx < query.rows && trainIdx < train.rows && distVal < bestDistance && mask(queryIdx, tra… in loop()
454 …__global__ void match(const PtrStepSz<T> query, const PtrStepSz<T> train, const Mask mask, int* be… in match() argument
466 …loop<BLOCK_SIZE, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, myBestDistance, myBestTr… in match()
483 void match(const PtrStepSz<T>& query, const PtrStepSz<T>& train, const Mask& mask, in match() argument
492 …match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, trainIdx.data, dist… in match()
517 const PtrStepSz<T> train = trains[imgIdx]; in match() local
519 …loop<BLOCK_SIZE, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, myBestDistance, myBest… in match()
559 void matchDispatcher(const PtrStepSz<T>& query, const PtrStepSz<T>& train, const Mask& mask, in matchDispatcher() argument
565 matchUnrolledCached<16, 64, Dist>(query, train, mask, trainIdx, distance, stream); in matchDispatcher()
569 matchUnrolledCached<16, 128, Dist>(query, train, mask, trainIdx, distance, stream); in matchDispatcher()
585 match<16, Dist>(query, train, mask, trainIdx, distance, stream); in matchDispatcher()
623 …template <typename T> void matchL1_gpu(const PtrStepSzb& query, const PtrStepSzb& train, const Ptr… in matchL1_gpu() argument
629 …Dist<T> >(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), SingleMask(mask), in matchL1_gpu()
635 … L1Dist<T> >(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), WithOutMask(), in matchL1_gpu()
648 …template <typename T> void matchL2_gpu(const PtrStepSzb& query, const PtrStepSzb& train, const Ptr… in matchL2_gpu() argument
654 …r<L2Dist>(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), SingleMask(mask), in matchL2_gpu()
660 …cher<L2Dist>(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), WithOutMask(), in matchL2_gpu()
673 …ename T> void matchHamming_gpu(const PtrStepSzb& query, const PtrStepSzb& train, const PtrStepSzb&… in matchHamming_gpu() argument
679 …mingDist>(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), SingleMask(mask), in matchHamming_gpu()
685 …HammingDist>(static_cast< PtrStepSz<T> >(query), static_cast< PtrStepSz<T> >(train), WithOutMask(), in matchHamming_gpu()