diff options
Diffstat (limited to 'include/armnn/Tensor.hpp')
-rw-r--r-- | include/armnn/Tensor.hpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/include/armnn/Tensor.hpp b/include/armnn/Tensor.hpp index 69ffbd9fcc..8814d89174 100644 --- a/include/armnn/Tensor.hpp +++ b/include/armnn/Tensor.hpp @@ -235,6 +235,17 @@ private: (m_QuantizationDim == other.m_QuantizationDim)); } + Quantization& operator=(const Quantization& other) + { + if(!(*this == other)) + { + m_Scales = other.m_Scales; + m_Offset = other.m_Offset; + m_QuantizationDim = other.m_QuantizationDim; + } + return *this; + } + std::vector<float> m_Scales; Optional<int32_t> m_Offset; Optional<unsigned int> m_QuantizationDim; |