diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2018-11-28 11:52:08 +0000 |
|---|---|---|
| committer | Mike Kelly <mike.kelly@arm.com> | 2018-11-28 11:52:08 +0000 |
| commit | 831faedd5d2f6306b23202fa4e450c6d241a44a0 (patch) | |
| tree | ad025f07969618a054144aa45333209399656cb5 /src/backends/cl/ClLayerSupport.cpp | |
| parent | 7e7261ed9a4f8d48c86e57044312e3df82189713 (diff) | |
| download | armnn-831faedd5d2f6306b23202fa4e450c6d241a44a0.tar.gz | |
IVGCVSW-1199: BATCH_TO_SPACE_ND integrate Arm Compute CL
Change-Id: Ic772acf481caed6e8a6b99f68a63dfc2a34c24dc
Diffstat (limited to 'src/backends/cl/ClLayerSupport.cpp')
| -rw-r--r-- | src/backends/cl/ClLayerSupport.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp index 218cc9b3d9..a877a5f25f 100644 --- a/src/backends/cl/ClLayerSupport.cpp +++ b/src/backends/cl/ClLayerSupport.cpp @@ -18,6 +18,7 @@ #include "workloads/ClAdditionWorkload.hpp" #include "workloads/ClActivationWorkload.hpp" #include "workloads/ClBatchNormalizationFloatWorkload.hpp" +#include "workloads/ClBatchToSpaceNdWorkload.hpp" #include "workloads/ClConvertFp16ToFp32Workload.hpp" #include "workloads/ClConvertFp32ToFp16Workload.hpp" #include "workloads/ClConvolution2dWorkload.hpp" @@ -165,6 +166,18 @@ bool ClLayerSupport::IsBatchNormalizationSupported(const TensorInfo& input, descriptor); } +bool ClLayerSupport::IsBatchToSpaceNdSupported(const TensorInfo& input, + const TensorInfo& output, + const BatchToSpaceNdDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported) const +{ + FORWARD_WORKLOAD_VALIDATE_FUNC(ClBatchToSpaceNdWorkloadValidate, + reasonIfUnsupported, + input, + output, + descriptor); +} + bool ClLayerSupport::IsConstantSupported(const TensorInfo& output, Optional<std::string&> reasonIfUnsupported) const { |