diff options
author | Tamás Nyíri <tamas.nyiri@arm.com> | 2021-10-26 14:47:57 +0100 |
---|---|---|
committer | Tamas Nyiri <tamas.nyiri@arm.com> | 2021-11-17 11:31:44 +0000 |
commit | 7b885b3cce70154596b1994b013ea91527117c26 (patch) | |
tree | cdc2ee30a6dc03a4e26e6783a84ccd9be867242a /src/backends/backendsCommon/WorkloadFactory.cpp | |
parent | 888a363115e0bf47f227c9db6fc1dbfe0418f69c (diff) | |
download | armnn-7b885b3cce70154596b1994b013ea91527117c26.tar.gz |
IVGCVSW-6509 Front End + Reference Workload implementation
Subtask of story: IVGCVSW-6164 Add a Pooling3d FrontEnd and Ref Implementation
* Add front end
* Add reference workload
* Add corresponding unit tests
Change-Id: Icce4146dd0a06a1da46a2def00a82d343e171750
Signed-off-by: Tamas Nyiri <tamas.nyiri@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactory.cpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index 55ce3554f9..ef2a34889e 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -831,6 +831,17 @@ bool IWorkloadFactory::IsLayerConfigurationSupported(const BackendId& backendId, reason); break; } + case LayerType::Pooling3d: + { + auto cLayer = PolymorphicDowncast<const Pooling3dLayer*>(&layer); + const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); + const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); + result = layerSupportObject.IsPooling3dSupported(OverrideDataType(input, dataType), + OverrideDataType(output, dataType), + cLayer->GetParameters(), + reason); + break; + } case LayerType::PreCompiled: { auto cLayer = PolymorphicDowncast<const PreCompiledLayer*>(&layer); @@ -1781,6 +1792,12 @@ std::unique_ptr<IWorkload> IWorkloadFactory::CreatePooling2d(const Pooling2dQueu return std::unique_ptr<IWorkload>(); } +std::unique_ptr<IWorkload> IWorkloadFactory::CreatePooling3d(const Pooling3dQueueDescriptor& /*descriptor*/, + const WorkloadInfo& /*info*/) const +{ + return std::unique_ptr<IWorkload>(); +} + std::unique_ptr<IWorkload> IWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const { |