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