diff options
| author | jimfly01 <jim.flynn@arm.com> | 2018-10-08 14:43:01 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
| commit | 6b0b53d307f956c3d67429f2a93df91f2eb2f483 (patch) | |
| tree | fcc03458f6396d9b9bee55674ca5af562ed5a16b /include | |
| parent | 0d9d4193200e8bfbaf4cb8f32c866d5fe73a3e7e (diff) | |
| download | armnn-6b0b53d307f956c3d67429f2a93df91f2eb2f483.tar.gz | |
IVGCVSW-1929: Fix for this defect (QASYM8 no scale)
* Now hand in an optional vector of strings to Optimize
function in which errors/warning messages are placed.
* Optimize function changed to check outputs of each
layer. If they are Quantized 8 bit but the scale has
not been set an error message is added for each such output.
* Unit test modelled on defect report added to exercise the fix.
!android-nn-driver:152483
Change-Id: If9901f5324a516f1ab62858266b38f98dae16201
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/INetwork.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 2c83909c83..aaf49a3d47 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -7,6 +7,7 @@ #include "armnn/NetworkFwd.hpp" #include "armnn/DescriptorsFwd.hpp" #include "armnn/TensorFwd.hpp" +#include "armnn/Optional.hpp" #include "armnn/Types.hpp" @@ -332,6 +333,7 @@ struct OptimizerOptions /// @param network INetwork description of the network to be optimized. /// @param backendPreferences The choice of the backend ordered by user preferences. /// @param deviceSpec DeviceSpec object as queried from the runtime. See IRuntime::GetDeviceSpec() +/// @param errMessages if there are failures or warnings a string describing same will be added to the vector /// @param options OptimizerOptions object with optimizer configuration options /// @return An IOptimizedNetworkPtr interface to the optimized network, throws an exception derived from /// armnn::Exception if process fails. @@ -339,5 +341,6 @@ struct OptimizerOptions IOptimizedNetworkPtr Optimize(const INetwork& network, const std::vector<armnn::Compute>& backendPreferences, const IDeviceSpec& deviceSpec, - const OptimizerOptions& options = OptimizerOptions()); + const OptimizerOptions& options = OptimizerOptions(), + Optional<std::vector<std::string>&> errMessages = EmptyOptional()); } //namespace armnn |