| // |
| // Copyright © 2017 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| #pragma once |
| |
| #include "TestLayerVisitor.hpp" |
| |
| namespace |
| { |
| |
| #define DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(name) \ |
| class Test##name##LayerVisitor : public armnn::TestLayerVisitor \ |
| { \ |
| public: \ |
| explicit Test##name##LayerVisitor(const char* layerName = nullptr) : armnn::TestLayerVisitor(layerName) {}; \ |
| \ |
| void Visit##name##Layer(const armnn::IConnectableLayer* layer, \ |
| const char* layerName = nullptr) override \ |
| { \ |
| CheckLayerPointer(layer); \ |
| CheckLayerName(layerName); \ |
| } \ |
| }; |
| |
| } // anonymous namespace |
| |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Addition) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Dequantize) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Division) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Floor) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Gather) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Maximum) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Merge) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Minimum) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Multiplication) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Prelu) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Quantize) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Subtraction) |
| DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Switch) |