diff options
author | giuros01 <giuseppe.rossini@arm.com> | 2018-09-13 09:31:40 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:55:19 +0000 |
commit | 188708170a51887868a7170c9ee963372bfb9ced (patch) | |
tree | fe8f1555022869d00a1c14a097fa58d013d9a8aa /arm_compute/core/Types.h | |
parent | 4e0ac6fd6bc647e811cc12c1e3d392a6151ec576 (diff) | |
download | ComputeLibrary-188708170a51887868a7170c9ee963372bfb9ced.tar.gz |
[COMPMID-1331] Add support for RoIAlign operator in CL
Change-Id: Ie215daacd10477309dbf8af1bb2b05b7a0a8f203
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/150773
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 4c81e1f700..9b81eccafb 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -808,17 +808,18 @@ private: }; /** ROI Pooling Layer Information class */ -class ROIPoolingLayerInfo +class ROIPoolingLayerInfo final { public: - /** Default Constructor + /** Constructor * - * @param[in] pooled_width Pooled width of the layer. - * @param[in] pooled_height Pooled height of the layer. - * @param[in] spatial_scale Spatial scale to be applied to the ROI coordinates and dimensions. + * @param[in] pooled_width Pooled width of the layer. + * @param[in] pooled_height Pooled height of the layer. + * @param[in] spatial_scale Spatial scale to be applied to the ROI coordinates and dimensions. + * @param[in] sampling_ratio Number of samples to include in each pooling region (if set to zero, a ceil(roi_dims/pooling_dims)) */ - ROIPoolingLayerInfo(unsigned int pooled_width, unsigned int pooled_height, float spatial_scale) - : _pooled_width(pooled_width), _pooled_height(pooled_height), _spatial_scale(spatial_scale) + ROIPoolingLayerInfo(unsigned int pooled_width, unsigned int pooled_height, float spatial_scale, unsigned int sampling_ratio = 0) + : _pooled_width(pooled_width), _pooled_height(pooled_height), _spatial_scale(spatial_scale), _sampling_ratio(sampling_ratio) { } /** Get the pooled width of the layer */ @@ -836,11 +837,17 @@ public: { return _spatial_scale; } + /** Get sampling ratio */ + unsigned int sampling_ratio() const + { + return _sampling_ratio; + } private: unsigned int _pooled_width; unsigned int _pooled_height; float _spatial_scale; + unsigned int _sampling_ratio; }; /** Activation Layer Information class */ |