aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/BaseIterator.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/BaseIterator.hpp')
-rw-r--r--src/backends/reference/workloads/BaseIterator.hpp15
1 files changed, 15 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/BaseIterator.hpp b/src/backends/reference/workloads/BaseIterator.hpp
index 97af95a0eb..ab6de2b37f 100644
--- a/src/backends/reference/workloads/BaseIterator.hpp
+++ b/src/backends/reference/workloads/BaseIterator.hpp
@@ -123,6 +123,21 @@ public:
}
};
+class ScaledInt32Decoder : public TypedIterator<const int32_t, Decoder<float>>
+{
+public:
+ ScaledInt32Decoder(const int32_t* data, const float scale)
+ : TypedIterator(data), m_Scale(scale) {}
+
+ float Get() const override
+ {
+ return static_cast<float>(*m_Iterator) * m_Scale;
+ }
+
+private:
+ const float m_Scale;
+};
+
class QASymm8Encoder : public TypedIterator<uint8_t, Encoder<float>>
{
public: