aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
Diffstat (limited to 'include')
-rw-r--r--include/armnn/Tensor.hpp11
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;