Searched refs:rois_ptr (Results 1 – 11 of 11) sorted by relevance
70 TRois *rois_ptr = static_cast<TRois *>(rois.data()); variable97 rois_ptr[values_per_roi * pw] = batch_idx;100 …rois_ptr[values_per_roi * pw + 1] = quantize_qasymm16(static_cast<float>(x1), rois.quantization_in…101 …rois_ptr[values_per_roi * pw + 2] = quantize_qasymm16(static_cast<float>(y1), rois.quantization_in…102 …rois_ptr[values_per_roi * pw + 3] = quantize_qasymm16(static_cast<float>(x2), rois.quantization_in…103 …rois_ptr[values_per_roi * pw + 4] = quantize_qasymm16(static_cast<float>(y2), rois.quantization_in…107 rois_ptr[values_per_roi * pw + 1] = static_cast<TRois>(x1);108 rois_ptr[values_per_roi * pw + 2] = static_cast<TRois>(y1);109 rois_ptr[values_per_roi * pw + 3] = static_cast<TRois>(x2);110 rois_ptr[values_per_roi * pw + 4] = static_cast<TRois>(y2);
69 uint16_t *rois_ptr = static_cast<uint16_t *>(rois.data()); variable96 rois_ptr[values_per_roi * pw] = batch_idx;97 rois_ptr[values_per_roi * pw + 1] = static_cast<uint16_t>(x1);98 rois_ptr[values_per_roi * pw + 2] = static_cast<uint16_t>(y1);99 rois_ptr[values_per_roi * pw + 3] = static_cast<uint16_t>(x2);100 rois_ptr[values_per_roi * pw + 4] = static_cast<uint16_t>(y2);
64 const auto *rois_ptr = reinterpret_cast<const uint16_t *>(rois.data()); in roi_pool_layer() local69 const unsigned int roi_batch = rois_ptr[values_per_roi * pw]; in roi_pool_layer()70 const auto x1 = rois_ptr[values_per_roi * pw + 1]; in roi_pool_layer()71 const auto y1 = rois_ptr[values_per_roi * pw + 2]; in roi_pool_layer()72 const auto x2 = rois_ptr[values_per_roi * pw + 3]; in roi_pool_layer()73 const auto y2 = rois_ptr[values_per_roi * pw + 4]; in roi_pool_layer()
148 const auto *rois_ptr = static_cast<const float *>(rois.data()); in roi_align_layer() local161 const unsigned int roi_batch = rois_ptr[values_per_roi * pw]; in roi_align_layer()162 const auto x1 = float(rois_ptr[values_per_roi * pw + 1]); in roi_align_layer()163 const auto y1 = float(rois_ptr[values_per_roi * pw + 2]); in roi_align_layer()164 const auto x2 = float(rois_ptr[values_per_roi * pw + 3]); in roi_align_layer()165 const auto y2 = float(rois_ptr[values_per_roi * pw + 4]); in roi_align_layer()
167 const auto *rois_ptr = reinterpret_cast<const uint16_t *>(_rois->buffer()); in run() local172 const unsigned int roi_batch = rois_ptr[values_per_roi * roi_indx]; in run()173 const auto x1 = rois_ptr[values_per_roi * roi_indx + 1]; in run()174 const auto y1 = rois_ptr[values_per_roi * roi_indx + 2]; in run()175 const auto x2 = rois_ptr[values_per_roi * roi_indx + 3]; in run()176 const auto y2 = rois_ptr[values_per_roi * roi_indx + 4]; in run()
236 const auto *rois_ptr = reinterpret_cast<const roi_data_type *>(rois->buffer()); in roi_align() local240 const unsigned int roi_batch = rois_ptr[values_per_roi * roi_indx]; in roi_align()242 roi_data_type qx1 = rois_ptr[values_per_roi * roi_indx + 1]; in roi_align()243 roi_data_type qy1 = rois_ptr[values_per_roi * roi_indx + 2]; in roi_align()244 roi_data_type qx2 = rois_ptr[values_per_roi * roi_indx + 3]; in roi_align()245 roi_data_type qy2 = rois_ptr[values_per_roi * roi_indx + 4]; in roi_align()
45 …* @param[out] rois_ptr Pointer to the rois. Supported data types: sam…
134 …* @param[in] rois_ptr Pointer to the ROIs tensor. Layout: { batch_ind…
114 …* @param[in] rois_ptr Pointer to the ROIs tensor. Layout: { batch_ind…
120 …* @param[in] rois_ptr Pointer to the ROIs tensor. Layout: { batch_ind…