diff options
| author | Jim Flynn <jim.flynn@arm.com> | 2019-05-02 11:33:25 +0100 |
|---|---|---|
| committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-08 16:10:59 +0000 |
| commit | f92dfced4498f12b9315c0fa377ba7be8998b607 (patch) | |
| tree | 4015208a5493ea414babba76c18bc72b6dbef875 /src/armnnQuantizer | |
| parent | a4247d5a50502811a6956dffd990c0254622b7e1 (diff) | |
| download | armnn-f92dfced4498f12b9315c0fa377ba7be8998b607.tar.gz | |
IVGCVSW-2833 Add Dynamic Quantization
Change-Id: Iba91e3f3625639f01d66f81a9f3e419e0e285d66
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnnQuantizer')
| -rw-r--r-- | src/armnnQuantizer/ArmNNQuantizerMain.cpp | 4 | ||||
| -rw-r--r-- | src/armnnQuantizer/QuantizationInput.hpp | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnQuantizer/ArmNNQuantizerMain.cpp b/src/armnnQuantizer/ArmNNQuantizerMain.cpp index 103597a72d..96d6515ba0 100644 --- a/src/armnnQuantizer/ArmNNQuantizerMain.cpp +++ b/src/armnnQuantizer/ArmNNQuantizerMain.cpp @@ -5,7 +5,7 @@ #include "CommandLineProcessor.hpp" #include <armnnDeserializer/IDeserializer.hpp> -#include <armnn/INetworkQuantizer.hpp> +#include <armnnQuantizer/INetworkQuantizer.hpp> #include <armnnSerializer/ISerializer.hpp> #include "QuantizationDataSet.hpp" #include "QuantizationInput.hpp" @@ -52,7 +52,7 @@ int main(int argc, char* argv[]) armnnQuantizer::InputLayerVisitor inputLayerVisitor; network->Accept(inputLayerVisitor); - for(armnnQuantizer::QuantizationInput quantizationInput : dataSet) + for (armnnQuantizer::QuantizationInput quantizationInput : dataSet) { armnn::InputTensors inputTensors; std::vector<std::vector<float>> inputData(quantizationInput.GetNumberOfInputs()); diff --git a/src/armnnQuantizer/QuantizationInput.hpp b/src/armnnQuantizer/QuantizationInput.hpp index ebabdd704f..1bfe84bc19 100644 --- a/src/armnnQuantizer/QuantizationInput.hpp +++ b/src/armnnQuantizer/QuantizationInput.hpp @@ -7,7 +7,7 @@ #include <map> #include <armnn/Types.hpp> -#include <armnn/INetworkQuantizer.hpp> +#include <armnnQuantizer/INetworkQuantizer.hpp> namespace armnnQuantizer { |