diff options
Diffstat (limited to 'src/armnn/test/TestNameOnlyLayerVisitor.hpp')
-rw-r--r-- | src/armnn/test/TestNameOnlyLayerVisitor.hpp | 82 |
1 files changed, 48 insertions, 34 deletions
diff --git a/src/armnn/test/TestNameOnlyLayerVisitor.hpp b/src/armnn/test/TestNameOnlyLayerVisitor.hpp index c0037ae28f..dec0d15a96 100644 --- a/src/armnn/test/TestNameOnlyLayerVisitor.hpp +++ b/src/armnn/test/TestNameOnlyLayerVisitor.hpp @@ -22,97 +22,97 @@ public: }; }; -class TestMultiplicationLayerVisitor : public TestLayerVisitor +class TestDivisionLayerVisitor : public TestLayerVisitor { public: - explicit TestMultiplicationLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestDivisionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitMultiplicationLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitDivisionLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestFloorLayerVisitor : public TestLayerVisitor +class TestEqualLayerVisitor : public TestLayerVisitor { public: - explicit TestFloorLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestEqualLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitFloorLayer(const IConnectableLayer* layer, + void VisitEqualLayer(const IConnectableLayer* layer, const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestDivisionLayerVisitor : public TestLayerVisitor +class TestFloorLayerVisitor : public TestLayerVisitor { public: - explicit TestDivisionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestFloorLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitDivisionLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitFloorLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestSubtractionLayerVisitor : public TestLayerVisitor +class TestGatherLayerVisitor : public TestLayerVisitor { public: - explicit TestSubtractionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestGatherLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitSubtractionLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitGatherLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestMaximumLayerVisitor : public TestLayerVisitor +class TestGreaterLayerVisitor : public TestLayerVisitor { public: - explicit TestMaximumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestGreaterLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitMaximumLayer(const IConnectableLayer* layer, + void VisitGreaterLayer(const IConnectableLayer* layer, const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestMinimumLayerVisitor : public TestLayerVisitor +class TestMultiplicationLayerVisitor : public TestLayerVisitor { public: - explicit TestMinimumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestMultiplicationLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitMinimumLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitMultiplicationLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestGreaterLayerVisitor : public TestLayerVisitor +class TestMaximumLayerVisitor : public TestLayerVisitor { public: - explicit TestGreaterLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestMaximumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitGreaterLayer(const IConnectableLayer* layer, + void VisitMaximumLayer(const IConnectableLayer* layer, const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -class TestEqualLayerVisitor : public TestLayerVisitor +class TestMinimumLayerVisitor : public TestLayerVisitor { public: - explicit TestEqualLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestMinimumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitEqualLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitMinimumLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; @@ -130,16 +130,30 @@ public: }; }; -class TestGatherLayerVisitor : public TestLayerVisitor +class TestSliceLayerVisitor : public TestLayerVisitor { public: - explicit TestGatherLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + explicit TestSliceLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; - void VisitGatherLayer(const IConnectableLayer* layer, - const char* name = nullptr) override { + void VisitSliceLayer(const IConnectableLayer* layer, + const SliceDescriptor& sliceDescriptor, + const char* name = nullptr) override + { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestSubtractionLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestSubtractionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitSubtractionLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { CheckLayerPointer(layer); CheckLayerName(name); }; }; -} //namespace armnn
\ No newline at end of file +} // namespace armnn |