• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.3
• source code
• scalar.h

×

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.scalar.h Maven / Gradle / Ivy

extern "C"
#include 
#include 
#include "deeplearning4j.h"
//scalar and current element
__device__ double op(double d1,double d2,double *params);

__device__ void transform(int n, int idx,double dx,double *dy,int incy,double *params,double *result) {
	int totalThreads = gridDim.x * blockDim.x;
	int tid = threadIdx.x;
	int i = blockIdx.x * blockDim.x + tid;

	for (; i < n; i += totalThreads) {
		result[i * incy] = op(dx,dy[i * incy],params);
	}

}