Lines Matching refs:Dist
92 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
94 … typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loopUnrolledCached()
99 Dist dist; in loopUnrolledCached()
125 typename Dist::result_type distVal = dist; in loopUnrolledCached()
138 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
145 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolledCached()
146 …typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * MAX_DESC_LEN… in matchUnrolledCached()
153 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_tra… in matchUnrolledCached()
169 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
179 …matchUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trai… in matchUnrolledCached()
186 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
194 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolledCached()
195 …typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * MAX_DESC_LEN… in matchUnrolledCached()
209 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_t… in matchUnrolledCached()
228 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
238 …matchUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trai… in matchUnrolledCached()
248 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
250 … typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loopUnrolled()
255 Dist dist; in loopUnrolled()
285 typename Dist::result_type distVal = dist; in loopUnrolled()
298 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
308 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolled()
309 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in matchUnrolled()
311 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, my… in matchUnrolled()
327 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
337 …matchUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, train, mas… in matchUnrolled()
344 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
356 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolled()
357 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in matchUnrolled()
365 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, … in matchUnrolled()
384 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
394 …matchUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trains, n,… in matchUnrolled()
404 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
406 typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loop()
411 Dist dist; in loop()
440 typename Dist::result_type distVal = dist; in loop()
453 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
463 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in match()
464 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in match()
466 …loop<BLOCK_SIZE, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, myBestDistance, myBestTr… in match()
482 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
492 …match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, trainIdx.data, dist… in match()
499 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
511 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in match()
512 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in match()
519 …loop<BLOCK_SIZE, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, myBestDistance, myBest… in match()
538 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
548 …match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, trains, n, mask, trainIdx.data, … in match()
558 template <typename Dist, typename T, typename Mask>
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()
589 template <typename Dist, typename T, typename Mask>
596 … matchUnrolledCached<16, 64, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream); in matchDispatcher()
600 … matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream); in matchDispatcher()
616 match<16, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream); in matchDispatcher()