diff options
Diffstat (limited to 'tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp')
-rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp index 742f968a7a..d92c17c5e5 100644 --- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp +++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp @@ -34,6 +34,8 @@ struct TensorPrinter void operator()(const std::vector<int>& values); + void operator()(const std::vector<int8_t>& values); + private: template<typename Container, typename Delegate> void ForEachValue(const Container& c, Delegate delegate); @@ -48,7 +50,8 @@ private: bool m_DequantizeOutput; }; -using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>>; +using TContainer = + mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>, std::vector<int8_t>>; using QuantizationParams = std::pair<float, int32_t>; void PopulateTensorWithData(TContainer& tensorData, |