kernels.double.norm2_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  double merge(double old,double opOutput,double *extraParams) {
       return opOutput + old;
 }

__global  double update(double old,double opOutput,double *extraParams) {
       return opOutput + old;
 }


__global  double op(double d1,double *extraParams) {
      return pow(d1,2);
}


__global  double postProcess(double reduction,int n,int xOffset,double *dx,int incx,double *params,double *result) {
             return sqrtf(reduction);
}
extern "C"
__kernel void norm2_strided_double(int n, int xOffset,double *dx,int incx,double *params,double *result) {
             transform(n,xOffset,dx,incx,params,result);
}