#ifndef OPENPOSE_NET_NMS_BASE_HPP
#define OPENPOSE_NET_NMS_BASE_HPP

#include <openpose/core/common.hpp>

namespace op
{
    template <typename T>
    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);

    template <typename T>
    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);

    template <typename T>
    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);
}

#endif // OPENPOSE_NET_NMS_BASE_HPP