All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
nmsBase.hpp
Go to the documentation of this file.
#ifndef OPENPOSE_NET_NMS_BASE_HPP
#define OPENPOSE_NET_NMS_BASE_HPP
#include <openpose/core/common.hpp>
namespace op
{
template <typename T>
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);
// Windows: Cuda functions do not include OP_API
template <typename T>
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);
// Windows: OpenCL functions do not include OP_API
template <typename T>
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
op::nmsOcl
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::nmsCpu
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)
common.hpp
op::nmsGpu
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)