26 #if defined(DATA_TYPE) && defined(SRC_DEPTH) && defined(STRIDE) 28 #define CALCULATE_SRC_COORDINATES(xo, yo, zo, xi, yi, zi) \ 30 int offset = zo / (int)SRC_DEPTH; \ 31 xi = xo * (int)STRIDE + offset % (int)STRIDE; \ 32 yi = yo * (int)STRIDE + offset / (int)STRIDE; \ 33 zi = zo % SRC_DEPTH; \ 59 __kernel
void reorg_layer_nchw(
65 int xo = get_global_id(0);
66 int yo = get_global_id(1);
67 int zo = get_global_id(2);
70 CALCULATE_SRC_COORDINATES(xo, yo, zo, xi, yi, zi);
72 int src_offset = xi *
sizeof(DATA_TYPE) + yi * src_stride_y + zi * src_stride_z;
73 *((__global DATA_TYPE *)out.
ptr) = *((__global DATA_TYPE *)(src_ptr + src_offset_first_element_in_bytes + src_offset));
75 #endif // // defined(DATA_TYPE) && defined(SRC_DEPTH) && defined(STRIDE) Structure to hold 3D tensor information.
SimpleTensor< float > src
#define CONVERT_TO_TENSOR3D_STRUCT(name)
__global uchar * ptr
Pointer to the starting postion of the buffer.
#define TENSOR3D_DECLARATION(name)