Sha256: a4eb8594a98132a1ac644243b1eaa4fc0ec3e0d6ac25646f0a89bca6af27c5dd
Contents?: true
Size: 277 Bytes
Versions: 13
Compression:
Stored size: 277 Bytes
Contents
% c_dtype = dtype_to_c_type(dtype)
__kernel void tanh_grad_<%= 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 - tanh(A[id]) * tanh(A[id]);
}
Version data entries
13 entries across 13 versions & 1 rubygems