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.hpp | |
| 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.hpp')
| -rw-r--r-- | src/backends/cl/ClLayerSupport.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp index b4d278995d..cd1b72286a 100644 --- a/src/backends/cl/ClLayerSupport.hpp +++ b/src/backends/cl/ClLayerSupport.hpp @@ -31,6 +31,11 @@ public: const BatchNormalizationDescriptor& descriptor, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; + bool IsBatchToSpaceNdSupported(const TensorInfo& input, + const TensorInfo& output, + const BatchToSpaceNdDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; + bool IsConstantSupported(const TensorInfo& output, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; |