ArmNN
NotReleased
TestNameOnlyLayerVisitor.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. 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
(
Gather
)
33
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Maximum)
34
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Merge)
35
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Minimum)
36
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Multiplication)
37
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Prelu)
38
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(
Quantize
)
39
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Subtraction)
40
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
(Switch)
armnn::Quantize
QuantizedType Quantize(float value, float scale, int32_t offset)
Explicit specialization of Quantize for int8_t.
Definition:
TypesUtils.cpp:31
armnn::Gather
void Gather(const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< float > ¶ms, const int32_t *indices, Encoder< float > &output)
Definition:
Gather.cpp:18
armnn::Dequantize
float Dequantize(QuantizedType value, float scale, int32_t offset)
Definition:
TypesUtils.cpp:47
TestLayerVisitor.hpp
DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS
#define DECLARE_TEST_NAME_ONLY_LAYER_VISITOR_CLASS(name)
Definition:
TestNameOnlyLayerVisitor.hpp:12
src
armnn
test
TestNameOnlyLayerVisitor.hpp
Generated on Fri Mar 13 2020 16:06:55 for ArmNN by
1.8.13