diff options
Diffstat (limited to 'include/armnn/Utils.hpp')
-rw-r--r-- | include/armnn/Utils.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/include/armnn/Utils.hpp b/include/armnn/Utils.hpp index b090651813..533117c5d9 100644 --- a/include/armnn/Utils.hpp +++ b/include/armnn/Utils.hpp @@ -5,6 +5,8 @@ #pragma once #include "armnn/TypesUtils.hpp" + +#include <mapbox/variant.hpp> #include <iostream> namespace armnn @@ -40,4 +42,9 @@ bool NeonDetected(); const std::string GetVersion(); +// Standard definition of TContainer used by ArmNN, use this definition or add alternative definitions here instead of +// defining your own. +using TContainer = + mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>, std::vector<int8_t>>; + } // namespace armnn |