diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-10-26 12:22:20 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-10-27 09:32:39 +0000 |
| commit | 615ad6cde5a56d8f38e1b9261621b40671a00ed7 (patch) | |
| tree | 2604a9e5711c864d1e8009a11ed4bff90c2d86fb /src/backends/cl/ClLayerSupport.hpp | |
| parent | ec5f7d13582d1e477dc3473223b503388092a352 (diff) | |
| download | armnn-615ad6cde5a56d8f38e1b9261621b40671a00ed7.tar.gz | |
IVGCVSW-6169 Add GpuAcc Conv3d Workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I8b73dccc14ef71cc083896102e24afb2e56e72e2
Diffstat (limited to 'src/backends/cl/ClLayerSupport.hpp')
| -rw-r--r-- | src/backends/cl/ClLayerSupport.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp index e7a6748f0a..e75aedaa6a 100644 --- a/src/backends/cl/ClLayerSupport.hpp +++ b/src/backends/cl/ClLayerSupport.hpp @@ -85,6 +85,13 @@ public: const Optional<TensorInfo>& biases, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; + bool IsConvolution3dSupported(const TensorInfo& input, + const TensorInfo& output, + const Convolution3dDescriptor& descriptor, + const TensorInfo& weights, + const Optional<TensorInfo>& biases, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; + bool IsDequantizeSupported(const TensorInfo& input, const TensorInfo& output, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; |