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

#include 

extern "C"
__kernel void uniform_float(int n,float lower,float upper,float *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;

          for(; i < n; i += totalThreads) {
              float u = result[i];
              result[i] = u * upper + (1 - u) * lower;
          }
}