diff options
Diffstat (limited to 'src/armnn/backends/RefLayerSupport.hpp')
| -rw-r--r-- | src/armnn/backends/RefLayerSupport.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/backends/RefLayerSupport.hpp b/src/armnn/backends/RefLayerSupport.hpp index fde09685ac..900cf6923a 100644 --- a/src/armnn/backends/RefLayerSupport.hpp +++ b/src/armnn/backends/RefLayerSupport.hpp @@ -51,6 +51,11 @@ bool IsDepthwiseConvolutionSupportedRef(const TensorInfo& input, const boost::optional<TensorInfo>& biases, std::string* reasonIfUnsupported = nullptr); +bool IsDivisionSupportedRef(const TensorInfo& input0, + const TensorInfo& input1, + const TensorInfo& output, + std::string* reasonIfUnsupported = nullptr); + bool IsFullyConnectedSupportedRef(const TensorInfo& input, const TensorInfo& output, const TensorInfo& weights, |