diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-06-22 15:58:32 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-06-30 13:00:52 +0000 |
commit | cbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09 (patch) | |
tree | eb6e5393726be21213e72f26a676b7c3809fc995 /src/backends/reference/workloads/BaseIterator.hpp | |
parent | 532a29d12d72f54549d8b71edd485c17af65698a (diff) | |
download | armnn-cbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09.tar.gz |
IVGCVSW-5007 Implement an Int32 reference Elementwise workload
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I6592169b74ac4294bc09647879aec0718c641f91
Diffstat (limited to 'src/backends/reference/workloads/BaseIterator.hpp')
-rw-r--r-- | src/backends/reference/workloads/BaseIterator.hpp | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/BaseIterator.hpp b/src/backends/reference/workloads/BaseIterator.hpp index be20644ab7..1f4f2da717 100644 --- a/src/backends/reference/workloads/BaseIterator.hpp +++ b/src/backends/reference/workloads/BaseIterator.hpp @@ -274,6 +274,21 @@ public: } }; +class Int32ToInt32tDecoder : public TypedIterator<const int32_t, Decoder<int32_t>> +{ +public: + Int32ToInt32tDecoder(const int32_t* data) + : TypedIterator(data){} + + Int32ToInt32tDecoder() + : Int32ToInt32tDecoder(nullptr) {} + + int32_t Get() const override + { + return *m_Iterator; + } +}; + class BooleanDecoder : public TypedIterator<const uint8_t, Decoder<float>> { public: @@ -470,6 +485,26 @@ public: } }; +class Int32ToInt32tEncoder : public TypedIterator<int32_t, Encoder<int32_t>> +{ +public: + Int32ToInt32tEncoder(int32_t* data) + : TypedIterator(data){} + + Int32ToInt32tEncoder() + : Int32ToInt32tEncoder(nullptr) {} + + void Set(int32_t right) override + { + *m_Iterator = right; + } + + int32_t Get() const override + { + return *m_Iterator; + } +}; + class BooleanEncoder : public TypedIterator<uint8_t, Encoder<bool>> { public: |