• Home
• org.nd4j
• nd4j-jocl-common
• 0.4-rc0
• source code
• transform.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.transform.h Maven / Gradle / Ivy

extern "C"
#include 
#include 
#include 
#include "deeplearning4j.h"
//an op for the kernel
__global  double op(double d1,double *params);

__global  void transform(int n,int idx,double *dy,int incy,double *params,double *result) {
	int totalThreads = get_num_groups(0) * get_local_size(0);
	int tid = get_local_id(0);
	int i = get_group_id(0) * get_local_size(0) + tid;
	/* equal, positive, non-unit increments. */
	for (; i < n; i += totalThreads) {
		       result[i * incy] = op(dy[i * incy],params);

	}


}