mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 22:44:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #17621 from YashasSamaga:cuda4dnn-mish-half

Browse Source
This commit is contained in:
Alexander Alekhin 2020-06-23 18:44:50 +00:00
commit 6fb46bb34b
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/dnn/src/cuda/functors.hpp
View File

@ -57,11 +57,19 @@ struct mish_functor<float> {
auto n = e * e + 2 * e; auto n = e * e + 2 * e;
if (value <= -0.6f) if (value <= -0.6f)
return value * fast_divide(n, n + 2); return value * fast_divide(n, n + 2);
return value - 2 * fast_divide(value, n + 2); return value - 2 * fast_divide(value, n + 2);
} }
}; };
#if !defined(__CUDA_ARCH__) || (__CUDA_ARCH__ >= 530)
template <>
struct mish_functor<__half> {
__device__ __half operator()(__half value) {
return mish_functor<float>()(value);
}
};
#endif
template <class T> template <class T>
struct sigmoid_functor { struct sigmoid_functor {
__device__ T operator()(T value) { __device__ T operator()(T value) {