diff options
author | David Monahan <David.Monahan@arm.com> | 2021-11-01 10:16:37 +0000 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2021-11-01 16:32:24 +0000 |
commit | 2d9956162dd002a41f7fb4fa6753195d33524c7f (patch) | |
tree | 35b88472ff1bc5374c40365785c626cfb1ddec2c /include/armnnUtils | |
parent | 6b9eba2f785093747f04af245da0cec7aca3931c (diff) | |
download | armnn-2d9956162dd002a41f7fb4fa6753195d33524c7f.tar.gz |
IVGCVSW-6359 Added support for Float16 (Half) to Execute Network
* Allows the user to specify float16 as a datatype
* Does not contain support for float16 on the TfLiteDelegate via
ExecuteNetwork
Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: Icba56feedab32662e2cf671cc46ada899cf40c6c
Diffstat (limited to 'include/armnnUtils')
-rw-r--r-- | include/armnnUtils/TContainer.hpp | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/include/armnnUtils/TContainer.hpp b/include/armnnUtils/TContainer.hpp index a55f9df488..d2a868ac18 100644 --- a/include/armnnUtils/TContainer.hpp +++ b/include/armnnUtils/TContainer.hpp @@ -6,15 +6,20 @@ #pragma once #include <armnn/TypesUtils.hpp> +#include <Half.hpp> #include <mapbox/variant.hpp> namespace armnnUtils { -// 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>>; +// Standard declaration of TContainer used by ArmNN +// Changes to this declaration constitute an api/abi break, new types should be added as a separate declaration and +// merged on the next planned api/abi update. +using TContainer = mapbox::util::variant<std::vector<float>, + std::vector<int>, + std::vector<uint8_t>, + std::vector<int8_t>, + std::vector<armnn::Half>>; } // namespace armnnUtils |