diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-16 14:27:45 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-17 09:03:43 +0000 |
commit | 636ab40d3741e12eaad11d5b50e4b34bfbb258b5 (patch) | |
tree | defaba57dc28c7b5dbe19075e24b6c8c0cefc9b2 /src/armnn/test/TestNameOnlyLayerVisitor.cpp | |
parent | 4dc64a69ba383ece509d442598617445a3b4847f (diff) | |
download | armnn-636ab40d3741e12eaad11d5b50e4b34bfbb258b5.tar.gz |
IVGCVSW-3875 Add frontend for SLICE layer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iebe675a0cee02db6f133d48ce58cbc1e233061db
Diffstat (limited to 'src/armnn/test/TestNameOnlyLayerVisitor.cpp')
-rw-r--r-- | src/armnn/test/TestNameOnlyLayerVisitor.cpp | 116 |
1 files changed, 73 insertions, 43 deletions
diff --git a/src/armnn/test/TestNameOnlyLayerVisitor.cpp b/src/armnn/test/TestNameOnlyLayerVisitor.cpp index 4bb9614385..c4c4a479eb 100644 --- a/src/armnn/test/TestNameOnlyLayerVisitor.cpp +++ b/src/armnn/test/TestNameOnlyLayerVisitor.cpp @@ -10,6 +10,7 @@ namespace armnn { BOOST_AUTO_TEST_SUITE(TestNameOnlyLayerVisitor) +// Addition BOOST_AUTO_TEST_CASE(CheckAdditionLayerVisitorName) { TestAdditionLayerVisitor visitor("AdditionLayer"); @@ -28,24 +29,45 @@ BOOST_AUTO_TEST_CASE(CheckAdditionLayerVisitorNameNullptr) layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckMultiplicationLayerVisitorName) +// Division +BOOST_AUTO_TEST_CASE(CheckDivisionLayerVisitorName) { - TestMultiplicationLayerVisitor visitor("MultiplicationLayer"); + TestDivisionLayerVisitor visitor("DivisionLayer"); Network net; - IConnectableLayer *const layer = net.AddMultiplicationLayer("MultiplicationLayer"); + IConnectableLayer *const layer = net.AddAdditionLayer("DivisionLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckMultiplicationLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckDivisionLayerVisitorNameNullptr) { - TestMultiplicationLayerVisitor visitor; + TestDivisionLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddMultiplicationLayer(); + IConnectableLayer *const layer = net.AddDivisionLayer(); + layer->Accept(visitor); +} + +// Equal +BOOST_AUTO_TEST_CASE(CheckEqualLayerVisitorName) +{ + TestEqualLayerVisitor visitor("EqualLayer"); + Network net; + + IConnectableLayer *const layer = net.AddEqualLayer("EqualLayer"); + layer->Accept(visitor); +} + +BOOST_AUTO_TEST_CASE(CheckEqualLayerVisitorNameNullptr) +{ + TestEqualLayerVisitor visitor; + Network net; + + IConnectableLayer *const layer = net.AddEqualLayer(); layer->Accept(visitor); } +// Floor BOOST_AUTO_TEST_CASE(CheckFloorLayerVisitorName) { TestFloorLayerVisitor visitor("FloorLayer"); @@ -64,42 +86,45 @@ BOOST_AUTO_TEST_CASE(CheckFloorLayerVisitorNameNullptr) layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckDivisionLayerVisitorName) +// Gather +BOOST_AUTO_TEST_CASE(CheckGatherLayerVisitorName) { - TestDivisionLayerVisitor visitor("DivisionLayer"); + TestGatherLayerVisitor visitor("GatherLayer"); Network net; - IConnectableLayer *const layer = net.AddAdditionLayer("DivisionLayer"); + IConnectableLayer *const layer = net.AddGatherLayer("GatherLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckDivisionLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckGatherLayerVisitorNameNullptr) { - TestDivisionLayerVisitor visitor; + TestGatherLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddDivisionLayer(); + IConnectableLayer *const layer = net.AddGatherLayer(); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckSubtractionLayerVisitorName) +// Greater +BOOST_AUTO_TEST_CASE(CheckGreaterLayerVisitorName) { - TestSubtractionLayerVisitor visitor("SubtractionLayer"); + TestGreaterLayerVisitor visitor("GreaterLayer"); Network net; - IConnectableLayer *const layer = net.AddSubtractionLayer("SubtractionLayer"); + IConnectableLayer *const layer = net.AddGreaterLayer("GreaterLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckSubtractionLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckGreaterLayerVisitorNameNullptr) { - TestSubtractionLayerVisitor visitor; + TestGreaterLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddSubtractionLayer(); + IConnectableLayer *const layer = net.AddGreaterLayer(); layer->Accept(visitor); } +// Maximum BOOST_AUTO_TEST_CASE(CheckMaximumLayerVisitorName) { TestMaximumLayerVisitor visitor("MaximumLayer"); @@ -118,6 +143,7 @@ BOOST_AUTO_TEST_CASE(CheckMaximumLayerVisitorNameNullptr) layer->Accept(visitor); } +// Minimum BOOST_AUTO_TEST_CASE(CheckMinimumLayerVisitorName) { TestMinimumLayerVisitor visitor("MinimumLayer"); @@ -136,78 +162,82 @@ BOOST_AUTO_TEST_CASE(CheckMinimumLayerVisitorNameNullptr) layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckGreaterLayerVisitorName) +// Multiplication +BOOST_AUTO_TEST_CASE(CheckMultiplicationLayerVisitorName) { - TestGreaterLayerVisitor visitor("GreaterLayer"); + TestMultiplicationLayerVisitor visitor("MultiplicationLayer"); Network net; - IConnectableLayer *const layer = net.AddGreaterLayer("GreaterLayer"); + IConnectableLayer *const layer = net.AddMultiplicationLayer("MultiplicationLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckGreaterLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckMultiplicationLayerVisitorNameNullptr) { - TestGreaterLayerVisitor visitor; + TestMultiplicationLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddGreaterLayer(); + IConnectableLayer *const layer = net.AddMultiplicationLayer(); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckEqualLayerVisitorName) +// Rsqrt +BOOST_AUTO_TEST_CASE(CheckRsqrtLayerVisitorName) { - TestEqualLayerVisitor visitor("EqualLayer"); + TestRsqrtLayerVisitor visitor("RsqrtLayer"); Network net; - IConnectableLayer *const layer = net.AddEqualLayer("EqualLayer"); + IConnectableLayer *const layer = net.AddRsqrtLayer("RsqrtLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckEqualLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckRsqrtLayerVisitorNameNullptr) { - TestEqualLayerVisitor visitor; + TestRsqrtLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddEqualLayer(); + IConnectableLayer *const layer = net.AddRsqrtLayer(); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckRsqrtLayerVisitorName) +// Slice +BOOST_AUTO_TEST_CASE(CheckSliceLayerVisitorName) { - TestRsqrtLayerVisitor visitor("RsqrtLayer"); + TestSliceLayerVisitor visitor("SliceLayer"); Network net; - IConnectableLayer *const layer = net.AddRsqrtLayer("RsqrtLayer"); + IConnectableLayer *const layer = net.AddSliceLayer(SliceDescriptor(), "SliceLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckRsqrtLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckSliceLayerVisitorNameNullptr) { - TestRsqrtLayerVisitor visitor; + TestSliceLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddRsqrtLayer(); + IConnectableLayer *const layer = net.AddSliceLayer(SliceDescriptor()); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckGatherLayerVisitorName) +// Subtraction +BOOST_AUTO_TEST_CASE(CheckSubtractionLayerVisitorName) { - TestGatherLayerVisitor visitor("GatherLayer"); + TestSubtractionLayerVisitor visitor("SubtractionLayer"); Network net; - IConnectableLayer *const layer = net.AddGatherLayer("GatherLayer"); + IConnectableLayer *const layer = net.AddSubtractionLayer("SubtractionLayer"); layer->Accept(visitor); } -BOOST_AUTO_TEST_CASE(CheckGatherLayerVisitorNameNullptr) +BOOST_AUTO_TEST_CASE(CheckSubtractionLayerVisitorNameNullptr) { - TestGatherLayerVisitor visitor; + TestSubtractionLayerVisitor visitor; Network net; - IConnectableLayer *const layer = net.AddGatherLayer(); + IConnectableLayer *const layer = net.AddSubtractionLayer(); layer->Accept(visitor); } BOOST_AUTO_TEST_SUITE_END() -} //namespace armnn
\ No newline at end of file +} // namespace armnn |