1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include "TestLayerVisitor.hpp" 8 9 namespace 10 { 11 12 #define DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(name) \ 13 class Test##name##LayerVisitor : public armnn::TestLayerVisitor \ 14 { \ 15 public: \ 16 explicit Test##name##LayerVisitor(const char* layerName = nullptr) : armnn::TestLayerVisitor(layerName) {}; \ 17 \ 18 void Visit##name##Layer(const armnn::IConnectableLayer* layer, \ 19 const char* layerName = nullptr) override \ 20 { \ 21 CheckLayerPointer(layer); \ 22 CheckLayerName(layerName); \ 23 } \ 24 }; 25 26 } // anonymous namespace 27 28 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Addition) 29 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Dequantize) 30 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Division) 31 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Floor) 32 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Maximum) 33 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Merge) 34 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Minimum) 35 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Multiplication) 36 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Prelu) 37 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Quantize) 38 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Rank) 39 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Subtraction) 40 DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(Switch) 41