diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/ILayerVisitor.hpp | 2 | ||||
-rw-r--r-- | include/armnn/TypesUtils.hpp | 4 | ||||
-rw-r--r-- | include/armnnQuantizer/INetworkQuantizer.hpp (renamed from include/armnn/INetworkQuantizer.hpp) | 0 |
3 files changed, 4 insertions, 2 deletions
diff --git a/include/armnn/ILayerVisitor.hpp b/include/armnn/ILayerVisitor.hpp index eabad58366..ab793bc587 100644 --- a/include/armnn/ILayerVisitor.hpp +++ b/include/armnn/ILayerVisitor.hpp @@ -347,6 +347,8 @@ public: virtual void VisitSwitchLayer(const IConnectableLayer* layer, const char* name = nullptr) = 0; + virtual void StartVisit() {} + virtual void FinishVisit() {} }; } // namespace armnn diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 837490d258..cb52471cd5 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -4,8 +4,8 @@ // #pragma once -#include "Tensor.hpp" -#include "Types.hpp" +#include <armnn/Tensor.hpp> +#include <armnn/Types.hpp> #include <cmath> #include <ostream> diff --git a/include/armnn/INetworkQuantizer.hpp b/include/armnnQuantizer/INetworkQuantizer.hpp index 89548d1057..89548d1057 100644 --- a/include/armnn/INetworkQuantizer.hpp +++ b/include/armnnQuantizer/INetworkQuantizer.hpp |