kernels.float.std_strided.kh Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nd4j-jocl-common Show documentation

There is a newer version: 0.4-rc3.8

#include "reduce.h"


__global  float merge(float f1,float f2,float *extraParams) {
   return f1 + f2;
}

__global  float update(float old,float opOutput,float *extraParams) {
       float mean = extraParams[2];
       float curr = pow(opOutput - mean,2.0);
       return old + curr;
 }


//an op for the kernel
__global  float op(float d1,float *extraParams) {
      return d1;

}

//post process result (for things like means etc)
__global  float postProcess(float reduction,int n,int xOffset,float *dx,int incx,float *extraParams,float *result) {
             float bias = extraParams[1];
            return  sqrtf((reduction - (pow(bias,2.0) / n)) / (float) (n - 1.0));

}

extern "C"
__kernel void std_strided_float(int n, int xOffset,float *dx,int incx,float *extraParams,float *result) {
              transform(n,xOffset,dx,incx,extraParams,result);

 }