From ac5aa1faef759e704d369e8164ea91c6d7049946 Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Wed, 5 Dec 2018 15:17:18 +0000 Subject: IVGCVSW-2315 Add descriptor for Debug Change-Id: I6b26c681989ef815752186f027a245ef8da7aea2 --- include/armnn/Descriptors.hpp | 15 +++++++++++++++ include/armnn/DescriptorsFwd.hpp | 1 + include/armnn/ILayerSupport.hpp | 1 + include/armnn/LayerSupport.hpp | 1 + 4 files changed, 18 insertions(+) (limited to 'include') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index c905eb2c97..8fbd32a76e 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -464,4 +464,19 @@ struct StridedSliceDescriptor DataLayout m_DataLayout; }; +struct DebugDescriptor +{ + DebugDescriptor() + : m_SlotIndex(0) + {} + + DebugDescriptor(const std::string name, unsigned int index) + : m_LayerName(name) + , m_SlotIndex(index) + {} + + std::string m_LayerName; + unsigned int m_SlotIndex; +}; + } diff --git a/include/armnn/DescriptorsFwd.hpp b/include/armnn/DescriptorsFwd.hpp index a2acc0717e..729f739b3d 100644 --- a/include/armnn/DescriptorsFwd.hpp +++ b/include/armnn/DescriptorsFwd.hpp @@ -27,6 +27,7 @@ struct SoftmaxDescriptor; struct OriginsDescriptor; struct ViewsDescriptor; struct StridedSliceDescriptor; +struct DebugDescriptor; using MergerDescriptor = OriginsDescriptor; using SplitterDescriptor = ViewsDescriptor; diff --git a/include/armnn/ILayerSupport.hpp b/include/armnn/ILayerSupport.hpp index 859eee25f8..2498aa945d 100644 --- a/include/armnn/ILayerSupport.hpp +++ b/include/armnn/ILayerSupport.hpp @@ -66,6 +66,7 @@ public: virtual bool IsDebugSupported(const TensorInfo& input, const TensorInfo& output, + const DebugDescriptor& descriptor, Optional reasonIfUnsupported = EmptyOptional()) const; virtual bool IsDepthwiseConvolutionSupported(const TensorInfo& input, diff --git a/include/armnn/LayerSupport.hpp b/include/armnn/LayerSupport.hpp index 45ba4979c9..53efc15649 100644 --- a/include/armnn/LayerSupport.hpp +++ b/include/armnn/LayerSupport.hpp @@ -82,6 +82,7 @@ bool IsConvolution2dSupported(const BackendId& backend, bool IsDebugSupported(const BackendId& backend, const TensorInfo& input, const TensorInfo& output, + const DebugDescriptor& descriptor, char* reasonIfUnsupported = nullptr, size_t reasonIfUnsupportedMaxLength = 1024); -- cgit v1.2.1