1 #ifndef OPENPOSE_NET_NMS_BASE_HPP
2 #define OPENPOSE_NET_NMS_BASE_HPP
9 OP_API void nmsCpu(T* targetPtr,
int* kernelPtr,
const T*
const sourcePtr,
const T threshold,
10 const std::array<int, 4>& targetSize,
const std::array<int, 4>& sourceSize,
11 const Point<T>& offset);
14 OP_API void nmsGpu(T* targetPtr,
int* kernelPtr,
const T*
const sourcePtr,
const T threshold,
15 const std::array<int, 4>& targetSize,
const std::array<int, 4>& sourceSize,
16 const Point<T>& offset);
19 OP_API void nmsOcl(T* targetPtr,
int* kernelPtr,
const T*
const sourcePtr,
const T threshold,
20 const std::array<int, 4>& targetSize,
const std::array<int, 4>& sourceSize,
21 const Point<T>& offset,
const int gpuID = 0);
24 #endif // OPENPOSE_NET_NMS_BASE_HPP
OP_API void nmsOcl(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset, const int gpuID=0)
OP_API void nmsGpu(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset)
OP_API void nmsCpu(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset)
#define OP_API
Definition: macros.hpp:18