diff options
Diffstat (limited to 'include/armnn/Tensor.hpp')
-rw-r--r-- | include/armnn/Tensor.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnn/Tensor.hpp b/include/armnn/Tensor.hpp index 57a243800e..d41cbb4b70 100644 --- a/include/armnn/Tensor.hpp +++ b/include/armnn/Tensor.hpp @@ -97,6 +97,8 @@ public: bool HasMultipleQuantizationScales() const { return m_Quantization.m_Scales.size() > 1; } + bool HasPerAxisQuantization() const; + std::vector<float> GetQuantizationScales() const; void SetQuantizationScales(const std::vector<float>& scales); |