@@ -316,18 +316,18 @@ __kernel void encode_center_size(__read_only image2d_t prior_box_image,
316316 output [3 ]);
317317}
318318
319- __kernel void decode_center_size (__read_only image2d_t prior_box_image ,
320- __read_only image2d_t target_box_image ,
321- __write_only image2d_t output_image ,
322- __private const int out_C ,
323- __private const int out_H ,
324- __private const int normalized ,
325- __private const float4 variance
319+ __kernel void decode_center_size_axis0 (__read_only image2d_t prior_box_image ,
320+ __read_only image2d_t target_box_image ,
321+ __write_only image2d_t output_image ,
322+ __private const int out_C ,
323+ __private const int out_H ,
324+ __private const int normalized ,
325+ __private const float4 variance
326326#ifdef PRIOR_BOX_VAR
327- ,
328- __read_only image2d_t prior_box_var_image
327+ ,
328+ __read_only image2d_t prior_box_var_image
329329#endif
330- ) {
330+ ) {
331331 const int out_c = get_global_id (0 );
332332 const int out_nh = get_global_id (1 );
333333 const int out_h = out_nh % out_H ;
0 commit comments