commit | 2241d18f16878ddef261eadda9a0a8f0672a60c8 | [log] [tgz] |
---|---|---|
author | Sadik Armagan <sadik.armagan@arm.com> | Fri Apr 23 15:20:07 2021 +0100 |
committer | Sadik Armagan <sadik.armagan@arm.com> | Wed Apr 28 09:00:47 2021 +0000 |
tree | 6ec9cfc5dae3f4435c00564e1c8c24054dc7f98c | |
parent | 5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5 [diff] [blame] |
IVGCVSW-5416 'Add android-nn-driver support for CAST Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: I02da912e5e4ca650b367ca40fe3f5ca5baa61cbb
diff --git a/src/backends/reference/workloads/RefCastWorkload.hpp b/src/backends/reference/workloads/RefCastWorkload.hpp index 6742ef0..870fb41 100644 --- a/src/backends/reference/workloads/RefCastWorkload.hpp +++ b/src/backends/reference/workloads/RefCastWorkload.hpp
@@ -18,6 +18,9 @@ public: using BaseWorkload<CastQueueDescriptor>::BaseWorkload; void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; +private: + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; }; } //namespace armnn