diff options
author | FrancisMurtagh <francis.murtagh@arm.com> | 2019-01-29 12:15:23 +0000 |
---|---|---|
committer | FrancisMurtagh <francis.murtagh@arm.com> | 2019-01-29 12:15:23 +0000 |
commit | 28d3d63cc0a33f8396b32fa8347c03912c065911 (patch) | |
tree | 6eaadaff9ec94479dff1834677971eecf7585ff6 /src/armnn/test/TestNameOnlyLayerVisitor.hpp | |
parent | 3d7efe9a21ad47f33d330240b3b901ad7d5a5a81 (diff) | |
download | armnn-28d3d63cc0a33f8396b32fa8347c03912c065911.tar.gz |
IVGCVSW-2549 Add name only layer unit tests for Visitor
Change-Id: I304ddd3efa2f1c5a8e2143276a97c031a624601e
Diffstat (limited to 'src/armnn/test/TestNameOnlyLayerVisitor.hpp')
-rw-r--r-- | src/armnn/test/TestNameOnlyLayerVisitor.hpp | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/src/armnn/test/TestNameOnlyLayerVisitor.hpp b/src/armnn/test/TestNameOnlyLayerVisitor.hpp new file mode 100644 index 0000000000..c0037ae28f --- /dev/null +++ b/src/armnn/test/TestNameOnlyLayerVisitor.hpp @@ -0,0 +1,145 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include "TestLayerVisitor.hpp" + +namespace armnn +{ + +// Concrete TestLayerVisitor subclasses for layers taking Name argument with overridden VisitLayer methods +class TestAdditionLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestAdditionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitAdditionLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestMultiplicationLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestMultiplicationLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitMultiplicationLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestFloorLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestFloorLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitFloorLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestDivisionLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestDivisionLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitDivisionLayer(const IConnectableLayer* layer, + 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); + }; +}; + +class TestMaximumLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestMaximumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitMaximumLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestMinimumLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestMinimumLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitMinimumLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestGreaterLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestGreaterLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitGreaterLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestEqualLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestEqualLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitEqualLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestRsqrtLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestRsqrtLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitRsqrtLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +class TestGatherLayerVisitor : public TestLayerVisitor +{ +public: + explicit TestGatherLayerVisitor(const char* name = nullptr) : TestLayerVisitor(name) {}; + + void VisitGatherLayer(const IConnectableLayer* layer, + const char* name = nullptr) override { + CheckLayerPointer(layer); + CheckLayerName(name); + }; +}; + +} //namespace armnn
\ No newline at end of file |