Sha256: 16c8f22df8faf7d85c42e38b07b331c872ac1b12783d098ab43c7541c2183438

Contents?: true

Size: 273 Bytes

Versions: 13

Compression:

Stored size: 273 Bytes

Contents

% c_dtype = dtype_to_c_type(dtype)

__kernel void sigmoid_<%= dtype %>(__global const <%= c_dtype %> *A, __global <%= c_dtype %> *C) {
    // Get the index of the current element to be processed
    const int id = get_global_id(0);

    C[id] = 1.0f/(1.0f + exp(-A[id]));
}

Version data entries

13 entries across 13 versions & 1 rubygems

Version Path
tensor_stream-opencl-0.3.2 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.3.1 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.3.0 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.10 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.9 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.8 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.6 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.5 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.4 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.3 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.2 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.1 lib/tensor_stream/opencl/kernels/sigmoid.cl
tensor_stream-opencl-0.2.0 lib/tensor_stream/opencl/kernels/sigmoid.cl