diff options
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 { |