Lines Matching refs:Dist
315 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
317 … typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loopUnrolledCached()
324 Dist dist; in loopUnrolledCached()
350 typename Dist::result_type distVal = dist; in loopUnrolledCached()
376 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
383 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolledCached()
384 …typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * MAX_DESC_LEN… in matchUnrolledCached()
393 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_tra… in matchUnrolledCached()
409 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
419 …matchUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trai… in matchUnrolledCached()
426 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
433 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolledCached()
434 …typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * MAX_DESC_LEN… in matchUnrolledCached()
451 …loopUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_t… in matchUnrolledCached()
470 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
480 …matchUnrolledCached<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trai… in matchUnrolledCached()
490 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
492 … typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loopUnrolled()
499 Dist dist; in loopUnrolled()
529 typename Dist::result_type distVal = dist; in loopUnrolled()
555 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
562 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolled()
563 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in matchUnrolled()
570 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, my… in matchUnrolled()
586 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
596 …matchUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, train, mas… in matchUnrolled()
603 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
610 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in matchUnrolled()
611 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in matchUnrolled()
626 …loopUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, … in matchUnrolled()
645 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
655 …matchUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, trains, n,… in matchUnrolled()
665 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
667 typename Dist::value_type* s_query, typename Dist::value_type* s_train, in loop()
674 Dist dist; in loop()
703 typename Dist::result_type distVal = dist; in loop()
729 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
736 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in match()
737 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in match()
744 …loop<BLOCK_SIZE, Dist>(queryIdx, query, 0, train, mask, s_query, s_train, myBestDistance1, myBestD… in match()
760 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
770 …match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, trainIdx.data, dist… in match()
777 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
784 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in match()
785 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in match()
800 …loop<BLOCK_SIZE, Dist>(queryIdx, query, imgIdx, train, m, s_query, s_train, myBestDistance1, myBes… in match()
819 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
829 …match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, trains, n, mask, trainIdx.data, … in match()
839 template <typename Dist, typename T, typename Mask>
846 …matchUnrolledCached<16, 64, Dist>(query, train, mask, static_cast< PtrStepSz<int2> >(trainIdx), st… in match2Dispatcher()
850 …matchUnrolledCached<16, 128, Dist>(query, train, mask, static_cast< PtrStepSz<int2> >(trainIdx), s… in match2Dispatcher()
866 …match<16, Dist>(query, train, mask, static_cast< PtrStepSz<int2> >(trainIdx), static_cast< PtrStep… in match2Dispatcher()
870 template <typename Dist, typename T, typename Mask>
877 …matchUnrolledCached<16, 64, Dist>(query, trains, n, mask, static_cast< PtrStepSz<int2> >(trainIdx)… in match2Dispatcher()
881 …matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, static_cast< PtrStepSz<int2> >(trainIdx… in match2Dispatcher()
897 …match<16, Dist>(query, trains, n, mask, static_cast< PtrStepSz<int2> >(trainIdx), static_cast< Ptr… in match2Dispatcher()
904 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
912 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in calcDistanceUnrolled()
913 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in calcDistanceUnrolled()
915 Dist dist; in calcDistanceUnrolled()
947 distVal = (typename Dist::result_type)dist; in calcDistanceUnrolled()
953 template <int BLOCK_SIZE, int MAX_DESC_LEN, typename Dist, typename T, typename Mask>
961 …calcDistanceUnrolled<BLOCK_SIZE, MAX_DESC_LEN, Dist><<<grid, block, smemSize, stream>>>(query, tra… in calcDistanceUnrolled()
968 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
976 typename Dist::value_type* s_query = (typename Dist::value_type*)(smem); in calcDistance()
977 … typename Dist::value_type* s_train = (typename Dist::value_type*)(smem + BLOCK_SIZE * BLOCK_SIZE); in calcDistance()
979 Dist dist; in calcDistance()
1010 distVal = (typename Dist::result_type)dist; in calcDistance()
1016 template <int BLOCK_SIZE, typename Dist, typename T, typename Mask>
1024 … calcDistance<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, train, mask, allDist); in calcDistance()
1034 template <typename Dist, typename T, typename Mask>
1041 calcDistanceUnrolled<16, 64, Dist>(query, train, mask, allDist, stream); in calcDistanceDispatcher()
1045 calcDistanceUnrolled<16, 128, Dist>(query, train, mask, allDist, stream); in calcDistanceDispatcher()
1061 calcDistance<16, Dist>(query, train, mask, allDist, stream); in calcDistanceDispatcher()
1133 template <typename Dist, typename T, typename Mask>
1140 match2Dispatcher<Dist>(query, train, mask, trainIdx, distance, stream); in matchDispatcher()
1144 calcDistanceDispatcher<Dist>(query, train, mask, allDist, stream); in matchDispatcher()