diff options
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2019-09-18 13:39:00 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-10-01 12:02:45 +0000 |
| commit | 2aa7fd011a4baff52dceb00a71b3674f819df8fc (patch) | |
| tree | 081a8b0a75ff130d2c6179acf1fe1f1b58943412 /tests/validation/reference/PoolingLayer.h | |
| parent | 5c4a8e96460eb83a6caef1c69ea5cbb4893858d7 (diff) | |
| download | ComputeLibrary-2aa7fd011a4baff52dceb00a71b3674f819df8fc.tar.gz | |
COMPMID-2601 [CL] add mixed precision support to PoolingLayer
* PoolingLayerInfo is updated with a new flag.
* CL Kernel is updated to use FP32 accumulation.
* CL pooling layer testscases are added for mixed precision.
* Reference pooling layer is updated to use FP32 accumulation.
Change-Id: I4ab2167cc7f86c86293cf50a0ca5119c04dc9c7e
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1973
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Diffstat (limited to 'tests/validation/reference/PoolingLayer.h')
| -rw-r--r-- | tests/validation/reference/PoolingLayer.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/validation/reference/PoolingLayer.h b/tests/validation/reference/PoolingLayer.h index 1c0b7ff40d..fc36d51c02 100644 --- a/tests/validation/reference/PoolingLayer.h +++ b/tests/validation/reference/PoolingLayer.h @@ -35,6 +35,8 @@ namespace validation { namespace reference { +template <typename T, typename ACC_T = T, typename std::enable_if<is_floating_point<T>::value, int>::type = 0> +SimpleTensor<T> pooling_layer_internal(const SimpleTensor<T> &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo); template <typename T> SimpleTensor<T> pooling_layer(const SimpleTensor<T> &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo); } // namespace reference |