diff options
author | David Monahan <david.monahan@arm.com> | 2019-04-25 16:03:38 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2019-04-29 07:15:23 +0000 |
commit | b85547024490d4c89538dcebfb0a2730e18a6d8d (patch) | |
tree | a6962f8c2392c4be7edd2b00249ff6a28bfdcf30 /include | |
parent | 04a8b05b25d3b752040a262a2725fa59753dd9b5 (diff) | |
download | armnn-b85547024490d4c89538dcebfb0a2730e18a6d8d.tar.gz |
IVGCVSW-2657: Fix to force correct quantisation parameters for QASYMM8 Softmax
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I0989ea843714ba1d5da756bb87ddefa3706b07eb
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/INetwork.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index a15ceb1c15..7141770298 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -219,6 +219,8 @@ public: const char* name = nullptr) = 0; /// Adds a softmax layer to the network. + /// If the data type is QAsymm8, then the output quantization parameters + /// must have a scale of 1/256 and an offset of 0 /// @param softmaxDescriptor - SoftmaxDescriptor to configure the softmax. /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. |