diff options
Diffstat (limited to 'tests/validation/NEON/UNIT/DynamicTensor.cpp')
-rw-r--r-- | tests/validation/NEON/UNIT/DynamicTensor.cpp | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/validation/NEON/UNIT/DynamicTensor.cpp b/tests/validation/NEON/UNIT/DynamicTensor.cpp index 731c6d1650..319aa6a8ee 100644 --- a/tests/validation/NEON/UNIT/DynamicTensor.cpp +++ b/tests/validation/NEON/UNIT/DynamicTensor.cpp @@ -36,7 +36,6 @@ #include "tests/framework/Macros.h" #include "tests/framework/datasets/Datasets.h" #include "tests/validation/fixtures/UNIT/DynamicTensorFixture.h" -#include <arm_compute/runtime/OffsetMemoryPool.h> namespace arm_compute { @@ -44,11 +43,19 @@ namespace test { namespace validation { +namespace +{ +using NENormLayerWrapper = SimpleFunctionWrapper<MemoryManagerOnDemand, NENormalizationLayer, ITensor>; +template <> +void NENormLayerWrapper::configure(arm_compute::ITensor *src, arm_compute::ITensor *dst) +{ + _func.configure(src, dst, NormalizationLayerInfo(NormType::CROSS_MAP, 3)); +} +} // namespace TEST_SUITE(NEON) TEST_SUITE(UNIT) TEST_SUITE(DynamicTensor) - -using NEDynamicTensorType3SingleFunction = DynamicTensorType3SingleFunction<Tensor, Accessor, Allocator, OffsetLifetimeManager, PoolManager, MemoryManagerOnDemand, NENormalizationLayer>; +using NEDynamicTensorType3SingleFunction = DynamicTensorType3SingleFunction<Tensor, Accessor, Allocator, OffsetLifetimeManager, PoolManager, MemoryManagerOnDemand, NENormLayerWrapper>; /** Tests the memory manager with dynamic input and output tensors. * |