diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-10-09 18:19:20 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:53 +0100 |
commit | 2400b6dd383f40febc78cf71475addd51ae9245f (patch) | |
tree | b5bf24014c236894dbaa6007cb44d2b4465d05d5 /src/armnn/test/CreateWorkload.hpp | |
parent | 9df2d951616e3d76b67a9852d5324de96633f0f9 (diff) | |
download | armnn-2400b6dd383f40febc78cf71475addd51ae9245f.tar.gz |
IVGCVSW-1967 Fix L2Norm NHWC unit test
* Applying the proper tensor shape during the tests depending on
the specific data layout used
Change-Id: I9c6c1e077236e84cecc1e10d7b2d0bd901df3ebd
Diffstat (limited to 'src/armnn/test/CreateWorkload.hpp')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index fed9dd6b33..b9735f4f2f 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -885,9 +885,14 @@ std::unique_ptr<L2NormalizationWorkload> CreateL2NormalizationWorkloadTest(armnn Layer* const input = graph.AddLayer<InputLayer>(0, "input"); Layer* const output = graph.AddLayer<OutputLayer>(0, "output"); + TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? + TensorShape{ 5, 20, 50, 67 } : TensorShape{ 5, 50, 67, 20 }; + TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? + TensorShape{ 5, 20, 50, 67 } : TensorShape{ 5, 50, 67, 20 }; + // Connects up. - armnn::TensorInfo inputTensorInfo({ 5, 20, 50, 67 }, DataType); - armnn::TensorInfo outputTensorInfo({ 5, 20, 50, 67 }, DataType); + armnn::TensorInfo inputTensorInfo(inputShape, DataType); + armnn::TensorInfo outputTensorInfo(outputShape, DataType); Connect(input, layer, inputTensorInfo); Connect(layer, output, outputTensorInfo); CreateTensorHandles(graph, factory); |