diff options
Diffstat (limited to 'src/armnn/NetworkQuantizerUtils.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizerUtils.hpp | 70 |
1 files changed, 0 insertions, 70 deletions
diff --git a/src/armnn/NetworkQuantizerUtils.hpp b/src/armnn/NetworkQuantizerUtils.hpp deleted file mode 100644 index 5497e1b898..0000000000 --- a/src/armnn/NetworkQuantizerUtils.hpp +++ /dev/null @@ -1,70 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include "NetworkQuantizationScheme.hpp" - -#include <armnn/Tensor.hpp> -#include <armnn/TypesUtils.hpp> -#include <armnn/ILayerVisitor.hpp> -#include <armnn/IStrategy.hpp> -#include <armnn/utility/Assert.hpp> - -#include <utility> -#include <limits> - -namespace armnn -{ - -template<typename srcType> -void QuantizeConstant(const srcType* src, uint8_t* dst, size_t numElements, float& scale, int& offset) -{ - ARMNN_ASSERT(src); - ARMNN_ASSERT(dst); - - float min = std::numeric_limits<srcType>::max(); - float max = std::numeric_limits<srcType>::lowest(); - for (size_t i = 0; i < numElements; ++i) - { - min = std::min(min, src[i]); - max = std::max(max, src[i]); - } - - QAsymmU8QuantizationScheme quantizationScheme; - OffsetScalePair qParams = quantizationScheme.ComputeScheme(min, max); - scale = qParams.first; - offset = qParams.second; - - for (size_t i = 0; i < numElements; ++i) - { - dst[i] = armnn::Quantize<uint8_t>(src[i], scale, offset); - } -} - -ConstTensor CreateQuantizedConst(const ConstTensor& tensor, std::vector<uint8_t>& backing); - -template <typename LayerContainer> -void VisitLayers(const LayerContainer& layerContainer, ILayerVisitor& visitor) -{ - visitor.StartVisit(); - for (auto layer : layerContainer) - { - layer->Accept(visitor); - } - visitor.FinishVisit(); -} - -template <typename LayerContainer> -void ApplyStrategyToLayers(const LayerContainer& layerContainer, IStrategy& strategy) -{ - for (auto layer : layerContainer) - { - layer->ExecuteStrategy(strategy); - } - strategy.FinishStrategy(); -} - -} // namespace armnn |