Sha256: e80169f5a4bcbd3b00d443fd80be9e0a2363f8f4b51a62abe551d1359f75e3ee
Contents?: true
Size: 467 Bytes
Versions: 13
Compression:
Stored size: 467 Bytes
Contents
% c_dtype = dtype_to_c_type(dtype)
__kernel void tanh_grad_<%= dtype %>(const int M, const int N, __global const <%= c_dtype %> *A, __global <%= c_dtype %> *C) {
// Get the index of the current element to be processed
const int globalRow = get_global_id(0); // Row ID of C (0..M)
const int globalCol = get_global_id(1); // Col ID of C (0..N)
C[globalRow * N + globalCol] = 1 - tanh(A[globalRow * N + globalCol]) * tanh(A[globalRow * N + globalCol]);
}
Version data entries
13 entries across 13 versions & 2 rubygems