fast sort of working

Joseph Redmon

2015-01-20 6e1d5b45de988bb795c4c505f22f2170a78b7746

1
2
3
4
5
__kernel void yoloswag420blazeit360noscope(__global float *input, __global float *rand, float prob, float scale, __global float *output)
{
    int id = get_global_id(0);
    output[id] = (rand[id] < prob) ? 0 : input[id]*scale;
}