Sha256: eeaec845c60120c368e0e94a53d60107c5ef2565b4a4c0bb4b7649db0aa4802f
Contents?: true
Size: 617 Bytes
Versions: 13
Compression:
Stored size: 617 Bytes
Contents
% c_dtype = dtype_to_c_type(dtype)
// same dimension add floating point op
__kernel void apply_adagrad_<%= dtype %>(
__global const <%= c_dtype %> *lr,
__global const <%= c_dtype %> *grad,
__global <%= c_dtype %> *output,
__global <%= c_dtype %> *acc
) {
// Get the index of the current element to be processed
const int index = get_global_id(0);
output[index] -= grad[index] * lr[0] * rsqrt(acc[index]);
}
Version data entries
13 entries across 13 versions & 1 rubygems