diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-10-18 10:33:23 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | a160b245a5c876d3630651e938a7c45ee30645be (patch) | |
tree | 0210f6e033a8c1043a0beffe12bf438a0d721302 /src/armnn | |
parent | e4dfd6ead59e17828f8814f0ecc5fa67f0c72868 (diff) | |
download | armnn-a160b245a5c876d3630651e938a7c45ee30645be.tar.gz |
IVGCVSW-2033 Add unit tests for the newly implemented NHWC support in
ref Normalization
* Added create workload unit tests for the NHWC data layout
Change-Id: Id4813105c027a26e277de64fecea1845fea39fd9
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index f3cf544fa3..51820a425f 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -517,9 +517,16 @@ std::unique_ptr<NormalizationWorkload> CreateNormalizationWorkloadTest(armnn::IW Layer* const input = graph.AddLayer<InputLayer>(0, "input"); Layer* const output = graph.AddLayer<OutputLayer>(0, "output"); + TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? + TensorShape{ 3, 5, 5, 1 } : TensorShape{ 3, 1, 5, 5 }; + TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? + TensorShape{ 3, 5, 5, 1 } : TensorShape{ 3, 1, 5, 5 }; + // Connects up. - Connect(input, layer, TensorInfo({3, 5, 5, 1}, DataType)); - Connect(layer, output, TensorInfo({3, 5, 5, 1}, DataType)); + armnn::TensorInfo inputTensorInfo(inputShape, DataType); + armnn::TensorInfo outputTensorInfo(outputShape, DataType); + Connect(input, layer, inputTensorInfo); + Connect(layer, output, outputTensorInfo); CreateTensorHandles(graph, factory); // Makes the workload and checks it. |