24 #if defined(DATA_TYPE) && defined(ACTIVATION_TYPE) && defined(NUM_CLASSES) && defined(VEC_SIZE) 28 #define SELECT_TYPE SELECT_VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE) 31 #define TYPE VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE) 60 __kernel
void yolo_layer_nchw(
76 const int box_ch_id = get_global_id(2) % (NUM_CLASSES + 5);
77 const bool activate = box_ch_id != 2 && box_ch_id != 3;
102 #else // VEC_SIZE != 1 131 __kernel
void yolo_layer_nhwc(
147 const int box_ch_id = get_global_id(0) % (NUM_CLASSES + 5);
148 const bool activate = box_ch_id != 2 && box_ch_id != 3;
171 #endif // VEC_SIZE != 1 172 #endif // defined(DATA_TYPE) && defined(ACTIVATION_TYPE) && defined(NUM_CLASSES) && defined(VEC_SIZE)
Structure to hold 3D tensor information.
#define CONVERT_TO_TENSOR3D_STRUCT(name)
SimpleTensor< T > select(const SimpleTensor< uint8_t > &c, const SimpleTensor< T > &x, const SimpleTensor< T > &y)
__global uchar * ptr
Pointer to the starting postion of the buffer.
#define TENSOR3D_DECLARATION(name)
#define ACTIVATION(op, DATA_TYPE, VEC_SIZE, x, A_VAL, B_VAL)