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 #9284 from ipuustin:dnn-opencl-fixes

Browse Source
This commit is contained in:
Alexander Alekhin 2017-08-01 13:06:00 +00:00
commit bab4bc0968
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/opencl/activations.cl
View File

@ -21,13 +21,13 @@ __kernel void TanHForward(const int count, __global T* in, __global T* out) {
__kernel void SigmoidForward(const int count, __global const T* in, __global T* out) {
int index = get_global_id(0);
if(index < count)
out[index] = 1. / (1. + exp(-in[index]));
out[index] = 1.0f / (1.0f + exp(-in[index]));
}
__kernel void BNLLForward(const int n, __global const T* in, __global T* out) {
int index = get_global_id(0);
if (index < n) {
out[index] = in[index] > 0 ? in[index] + log(1. + exp(-in[index])) : log(1. + exp(in[index]));
out[index] = in[index] > 0 ? in[index] + log(1.0f + exp(-in[index])) : log(1.0f + exp(in[index]));
}
}
@ -41,4 +41,4 @@ __kernel void PowForward(const int n, __global const T* in, __global T* out, con
int index = get_global_id(0);
if (index < n)
out[index] = pow(shift + scale * in[index], power);
}
}