diff options
Diffstat (limited to 'include/armnn/ILayerSupport.hpp')
| -rw-r--r-- | include/armnn/ILayerSupport.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/include/armnn/ILayerSupport.hpp b/include/armnn/ILayerSupport.hpp index c67569bf00..d168226402 100644 --- a/include/armnn/ILayerSupport.hpp +++ b/include/armnn/ILayerSupport.hpp @@ -14,6 +14,7 @@ #include <functional> #include <memory> #include <vector> +#include "ArmNN.hpp" namespace armnn { @@ -41,6 +42,11 @@ public: const TensorInfo& output, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0; + virtual bool IsArgMinMaxSupported(const TensorInfo& input, + const TensorInfo& output, + const ArgMinMaxDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0; + virtual bool IsBatchNormalizationSupported(const TensorInfo& input, const TensorInfo& output, const TensorInfo& mean, |