diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/NetworkQuantizer.cpp | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/src/armnn/NetworkQuantizer.cpp b/src/armnn/NetworkQuantizer.cpp index f308d54d49..4a8bae52f2 100644 --- a/src/armnn/NetworkQuantizer.cpp +++ b/src/armnn/NetworkQuantizer.cpp @@ -3,36 +3,34 @@ // SPDX-License-Identifier: MIT // -#include <armnn/ILayerVisitor.hpp> -#include <armnn/INetwork.hpp> -#include <armnn/Tensor.hpp> -#include <armnn/Types.hpp> -#include <TensorUtils.hpp> -#include <TensorIOUtils.hpp> - +#include "NetworkQuantizer.hpp" +#include "NetworkQuantizerUtils.hpp" #include "Graph.hpp" #include "Layer.hpp" #include "Network.hpp" -#include "NetworkQuantizer.hpp" -#include "NetworkQuantizerUtils.hpp" - #include "DynamicQuantizationVisitor.hpp" #include "StaticRangeVisitor.hpp" #include "QuantizerVisitor.hpp" #include "OverrideInputRangeVisitor.hpp" -#include <vector> -#include <cmath> +#include <armnn/ILayerVisitor.hpp> +#include <armnn/INetwork.hpp> +#include <armnn/Tensor.hpp> +#include <armnn/Types.hpp> + +#include <TensorUtils.hpp> +#include <TensorIOUtils.hpp> #include <boost/variant.hpp> +#include <vector> +#include <cmath> namespace armnn { using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>>; - INetworkQuantizer* INetworkQuantizer::CreateRaw(INetwork* inputNetwork, const QuantizerOptions& options) { return new NetworkQuantizer(inputNetwork, options); |