• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.4
• source code
• mean_strided.cu

×

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.mean_strided.cu Maven / Gradle / Ivy

#include "reduce.h"



__device__ double merge(double old,double opOutput,double *extraParams) {
       return opOutput + old;
 }

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

__device__ double op(double d1,double *extraParams) {
       return d1;
}



__device__ double postProcess(double reduction,int n,int xOffset,double *dx,int incx,double *extraParams,double *result) {
             return reduction / (double) n;
}


extern "C"
__global__ void mean_strided_double(int n, int xOffset,double *dx,int incx,double *extraParams,double *result) {
             transform(n,xOffset,dx,incx,extraParams,result);
}