• Home
• org.nd4j
• nd4j-jocl-common
• 0.4-rc0
• source code
• cosinesimilarity_strided.kh

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

kernels.double.cosinesimilarity_strided.kh Maven / Gradle / Ivy

#include "reduce3.h"


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


/**
 An op on the device
 @param d1 the first operator
 @param d2 the second operator
*/
__global  double op(double d1,double d2,double *extraParams) {
      return d1 * d2;
}


//post process result (for things like means etc)
__global  double postProcess(double reduction,int n,int xOffset,double *dx,int incx,double *extraParams,double *result) {
            return reduction / extraParams[0] / extraParams[1];
}

extern "C"
__kernel void cosinesimilarity_strided_double(int n, int xOffset,int yOffset,double *dx,double *dy,int incx,int incy,double *extraParams,double *result) {
              transform_pair(n,xOffset,yOffset,dx,dy,incx,incy,extraParams,result);

 }