|
|
|
@@ -69,7 +69,7 @@ __kernel void Scale_H_IMG(__read_only image2d_t input, __read_only image2d_t sca |
|
|
|
__write_only image2d_t output, const int2 output_shape, const int H, const int act_type) { |
|
|
|
int X = get_global_id(0); |
|
|
|
int Y = get_global_id(1); |
|
|
|
if (X >= output_shape.x || Y >= output_shape.y) { |
|
|
|
if (X >= output_shape.x || Y >= output_shape.y || H == 0) { |
|
|
|
return; |
|
|
|
} |
|
|
|
int h = Y % H; |
|
|
|
|