diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-06-23 16:22:23 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-06-23 18:07:36 +0000 |
commit | a792a05ba521c426462eb050f96a98b1ad156c58 (patch) | |
tree | c40cec3fa0352b20fec9d42b6e8702edf7a7aece /src/backends/neon/NeonWorkloadFactory.cpp | |
parent | 4c10dfc213a59ec7bbf56b645e313083638b81a4 (diff) | |
download | armnn-a792a05ba521c426462eb050f96a98b1ad156c58.tar.gz |
IVGCVSW-4622 Add NEON FILL Workload
* Added Neon workload for Fill Operator
* Enabled Fill operator tests on Neon
* NEFill function does not have validate() function yet
IsLayerSupported() function return true at the moment
* Added INT32 supported type for CpuRef
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I17bf5ec13750f46322a30653e15ba2a514f61f08
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.cpp')
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index 82934c3b25..34f8445f95 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -265,6 +265,12 @@ std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateEqual(const EqualQueueDesc return CreateComparison(comparisonDescriptor, info); } +std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateFill(const FillQueueDescriptor& descriptor, + const WorkloadInfo& info) const +{ + return std::make_unique<NeonFillWorkload>(descriptor, info); +} + std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateFloor(const FloorQueueDescriptor& descriptor, const WorkloadInfo& info) const { |