diff options
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index d3ce7ab62c..9d00a69518 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -574,8 +574,10 @@ bool CheckScaleSetOnQuantizedType(Layer* layer, Optional<std::vector<std::string for (unsigned int i = 0; i < numOutputs; i++) { OutputSlot& outputSlot = layer->GetOutputSlot(i); TensorInfo info = outputSlot.GetTensorInfo(); - if (DataType::QAsymmU8 == info.GetDataType()) { - if (0.f == info.GetQuantizationScale()) { + if (DataType::QAsymmU8 == info.GetDataType()) + { + if (0.f == info.GetQuantizationScale()) + { noErrors = false; std::stringstream ss; ss << "output " << i << " of layer " << GetLayerTypeAsCString(layer->GetType()) |