Go to the source code of this file.
|
template<typename input_data_type > |
input_data_type | roi_align_1x1 (const ITensor *input, unsigned int roi_batch, float region_start_x, float bin_size_x, int grid_size_x, float region_end_x, float region_start_y, float bin_size_y, int grid_size_y, float region_end_y, int pz) |
| Average pooling over an aligned window. More...
|
|
template<typename input_data_type > |
input_data_type | roi_align_1x1_qasymm8 (const ITensor *input, unsigned int roi_batch, float region_start_x, float bin_size_x, int grid_size_x, float region_end_x, float region_start_y, float bin_size_y, int grid_size_y, float region_end_y, int pz, const QuantizationInfo &out_qinfo) |
| Average pooling over an aligned window. More...
|
|
float | compute_region_coordinate (int p, float bin_size, float roi_anchor, float max_value) |
|
template<typename input_data_type , typename roi_data_type > |
void | roi_align (const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, const Window &window, const ThreadInfo &info) |
|