aboutsummaryrefslogtreecommitdiff
path: root/include/armnn
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn')
-rw-r--r--include/armnn/Descriptors.hpp15
-rw-r--r--include/armnn/DescriptorsFwd.hpp1
-rw-r--r--include/armnn/ILayerSupport.hpp1
-rw-r--r--include/armnn/LayerSupport.hpp1
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);