aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/LayerValidateOutputTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/LayerValidateOutputTest.cpp')
-rw-r--r--src/armnn/test/LayerValidateOutputTest.cpp84
1 files changed, 0 insertions, 84 deletions
diff --git a/src/armnn/test/LayerValidateOutputTest.cpp b/src/armnn/test/LayerValidateOutputTest.cpp
deleted file mode 100644
index d47959cb65..0000000000
--- a/src/armnn/test/LayerValidateOutputTest.cpp
+++ /dev/null
@@ -1,84 +0,0 @@
-//
-// Copyright © 2017 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-#include <armnn/ArmNN.hpp>
-
-#include <Graph.hpp>
-#include <layers/BatchToSpaceNdLayer.hpp>
-#include <layers/SpaceToDepthLayer.hpp>
-
-#include <boost/algorithm/string.hpp>
-#include <boost/test/unit_test.hpp>
-
-BOOST_AUTO_TEST_SUITE(LayerValidateOutput)
-
-BOOST_AUTO_TEST_CASE(TestBatchToSpaceInferOutputShape)
-{
- armnn::Graph graph;
-
- armnn::BatchToSpaceNdDescriptor descriptor;
- descriptor.m_BlockShape = {2, 2};
- descriptor.m_Crops = {{0, 0}, {2, 0}};
- descriptor.m_DataLayout = armnn::DataLayout::NHWC;
-
- armnn::BatchToSpaceNdLayer* const batchToSpaceLayer =
- graph.AddLayer<armnn::BatchToSpaceNdLayer>(descriptor, "batchToSpace");
-
- std::vector<armnn::TensorShape> shapes;
- const std::vector<unsigned int> theDimSizes = {8, 1, 3, 1};
- armnn::TensorShape shape(4, theDimSizes.data());
- shapes.push_back(shape);
-
- const std::vector<unsigned int> expectedDimSizes = {2, 2, 4, 1};
- armnn::TensorShape expectedShape(4, expectedDimSizes.data());
-
- BOOST_CHECK(expectedShape == batchToSpaceLayer->InferOutputShapes(shapes).at(0));
-}
-
-BOOST_AUTO_TEST_CASE(TestSpaceToDepthInferOutputShape)
-{
- armnn::Graph graph;
-
- armnn::SpaceToDepthDescriptor descriptor;
- descriptor.m_BlockSize = 2;
- descriptor.m_DataLayout = armnn::DataLayout::NHWC;
-
- armnn::SpaceToDepthLayer* const spaceToDepthLayer =
- graph.AddLayer<armnn::SpaceToDepthLayer>(descriptor, "spaceToDepth");
-
- std::vector<armnn::TensorShape> shapes;
- const std::vector<unsigned int> dimSizes{ 1, 16, 8, 3 };
- armnn::TensorShape shape(4, dimSizes.data());
- shapes.push_back(shape);
-
- const std::vector<unsigned int> expectedDimSizes{ 1, 8, 4, 12 };
- armnn::TensorShape expectedShape(4, expectedDimSizes.data());
-
- BOOST_CHECK(expectedShape == spaceToDepthLayer->InferOutputShapes(shapes).at(0));
-}
-
-BOOST_AUTO_TEST_CASE(TestPreluInferOutputShape)
-{
- armnn::Graph graph;
-
- armnn::PreluLayer* const preluLayer = graph.AddLayer<armnn::PreluLayer>("prelu");
-
- std::vector<armnn::TensorShape> inputShapes
- {
- { 4, 1, 2 }, // Input shape
- { 5, 4, 3, 1} // Alpha shape
- };
-
- const std::vector<armnn::TensorShape> expectedOutputShapes
- {
- { 5, 4, 3, 2 } // Output shape
- };
-
- const std::vector<armnn::TensorShape> outputShapes = preluLayer->InferOutputShapes(inputShapes);
-
- BOOST_CHECK(outputShapes.size() == 1);
- BOOST_CHECK(outputShapes[0] == expectedOutputShapes[0]);
-}
-
-BOOST_AUTO_TEST_SUITE_END()