commit 2605b236d103e1ba27069e0d668599042a4761af
author Finn Williams <Finn.Williams@arm.com> Wed Jun 10 15:53:46 2020 +0100
committer Finn Williams <Finn.Williams@arm.com> Mon Jul 06 21:50:50 2020 +0100
tree dc0a408acb5749a1f51544fd5857452377d05aab
parent c5789ca2e432075e2c92e7e0d99139c5329280e6

IVGCVSW-4624 Add a RANK Reference Implementation

 * Add Rank front end
 * Add Rank reference implementation
 * Add Rank serialization support
 * Add Scalar serialization support

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I06e4a468c2a84e79bae2e6c5348596bbbf853b4b

src/backends/reference/RefLayerSupport.hpp

diff --git a/src/backends/reference/RefLayerSupport.hpp b/src/backends/reference/RefLayerSupport.hpp
index 7d2bbf2..a233082 100644
--- a/src/backends/reference/RefLayerSupport.hpp
+++ b/src/backends/reference/RefLayerSupport.hpp
@@ -265,6 +265,10 @@
                           const LstmInputParamsInfo& paramsInfo,
                           Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsRankSupported(const TensorInfo& input,
+                         const TensorInfo& output,
+                         Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
     bool IsReshapeSupported(const TensorInfo& input,
                             const TensorInfo& output,
                             const ReshapeDescriptor& descriptor,