diff options
Diffstat (limited to 'src/armnn/Tensor.cpp')
| -rw-r--r-- | src/armnn/Tensor.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Tensor.cpp b/src/armnn/Tensor.cpp index f4b8b509b6..dad9722aeb 100644 --- a/src/armnn/Tensor.cpp +++ b/src/armnn/Tensor.cpp @@ -230,6 +230,11 @@ bool TensorInfo::IsTypeSpaceMatch(const TensorInfo& other) const return match; } +bool TensorInfo::HasPerAxisQuantization() const +{ + return HasMultipleQuantizationScales() || m_Quantization.m_QuantizationDim.has_value(); +} + std::vector<float> TensorInfo::GetQuantizationScales() const { return m_Quantization.m_Scales; |