• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.4
• source code
• setrange_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.setrange_strided.cu Maven / Gradle / Ivy

#include "transform.h"


__device__ double op(double d1,double *params) {
       double min = params[0];
       double max = params[1];
    if(d1 >= min && d1 <= max)
               return d1;
           if(min == 0 && max == 1) {
               double val = 1 / (1 + exp(-d1));
               return (floor(val * (max - min)) + min);
           }

           double ret =  (floor(d1 * (max - min)) + min);
           return ret;
     
}

extern "C"
__global__ void setrange_strided_double(int n,int idx,double *dy,int incy,double *params,double *result) {
       transform(n,idx,dy,incy,params,result);

 }