diff options
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 |