Sha256: a22b6d9818e751d8ac12bb6d107520dbcbbab92b6e7bfa022a5bb3217331e8f2
Contents?: true
Size: 331 Bytes
Versions: 10
Compression:
Stored size: 331 Bytes
Contents
% c_dtype = dtype_to_c_type(dtype)
__kernel void relu6_<%= 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] = min((<%= c_dtype %>)max((<%= c_dtype %>) A[id], (<%= c_dtype %>)0), (<%= c_dtype %>)6);
}
Version data entries
10 entries across 10 versions & 1 rubygems