diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-20 12:31:24 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-22 11:17:19 +0000 |
| commit | 9d3a831d4131f8a8b37f127f11d36848d33e8496 (patch) | |
| tree | 5ba1ab6ee69da474175ab82eda7ead31298af4c6 /tests/validation/reference/NormalizationLayer.cpp | |
| parent | 3112e33d8d0b987e85107390a0350bd5988f5f01 (diff) | |
| download | ComputeLibrary-9d3a831d4131f8a8b37f127f11d36848d33e8496.tar.gz | |
COMPMID-1648: CLNormalizationLayer IN_MAP_2D support for NHWC for FP32/FP16
Change-Id: I49f1d865f5e7562f1d80db849353a89ef77e6a9e
Diffstat (limited to 'tests/validation/reference/NormalizationLayer.cpp')
| -rw-r--r-- | tests/validation/reference/NormalizationLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/NormalizationLayer.cpp b/tests/validation/reference/NormalizationLayer.cpp index e6ca233e75..d57e6f15a9 100644 --- a/tests/validation/reference/NormalizationLayer.cpp +++ b/tests/validation/reference/NormalizationLayer.cpp @@ -56,7 +56,7 @@ SimpleTensor<T> normalization_layer(const SimpleTensor<T> &src, NormalizationLay // IN_MAP_1D and CROSS_MAP normalize over a single axis only int radius_rows = (NormType::IN_MAP_2D == type) ? norm_size / 2 : 0; - if(type == NormType::CROSS_MAP) + if(info.is_cross_map()) { // Remove also depth from upper dimensions since it is the dimension we // want to use for normalization |