diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-12-05 15:17:18 +0000 |
|---|---|---|
| committer | Les Bell <les.bell@arm.com> | 2018-12-06 08:36:56 +0000 |
| commit | ac5aa1faef759e704d369e8164ea91c6d7049946 (patch) | |
| tree | 4202198355c149afdaa72e20e3a299e8b8974256 /include | |
| parent | 8174f3629d4b0dc0af7a75d2a201d756bc9e9d5b (diff) | |
| download | armnn-ac5aa1faef759e704d369e8164ea91c6d7049946.tar.gz | |
IVGCVSW-2315 Add descriptor for Debug
Change-Id: I6b26c681989ef815752186f027a245ef8da7aea2
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 15 | ||||
| -rw-r--r-- | include/armnn/DescriptorsFwd.hpp | 1 | ||||
| -rw-r--r-- | include/armnn/ILayerSupport.hpp | 1 | ||||
| -rw-r--r-- | include/armnn/LayerSupport.hpp | 1 |
4 files changed, 18 insertions, 0 deletions
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<std::string&> 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); |