commit 15eb5832f45d35c5041ba35a43787e8003e22edb
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Mon May 20 15:31:05 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Thu May 23 13:37:29 2019 +0000
tree 09fed880bfb9f384d3170aad5c76e4d565267e20
parent 495852f2adef1d11fbf13ce6347cf61973ce1a65

IVGCVSW-2771 Fix SubTensor error in vgg16 ExecuteNetwork NEON

 * Add check if Sub-tensors cannot be used, call ACL function
 * Add computation of SplitAxis from SplitterDescriptor
 * Add NeonSplitterWorkload functions
 * Modify IsSplitterSupported to call ACL validate function
if sub-tensor cannot be used
 * Also check if quantization parameters match when using sub-tensors
 * Add more unit tests for Splitter in TfParser and TfLiteParser

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I31e4c7d055117c83c65b598c4125442173242226

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 7beff72..6ad6816 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -998,6 +998,19 @@
                                      &TrueFunc<>);
 }
 
+bool RefLayerSupport::IsSplitterSupported(const TensorInfo& input,
+                                          const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
+                                          const ViewsDescriptor& descriptor,
+                                          Optional<std::string&> reasonIfUnsupported) const
+{
+    ignore_unused(descriptor);
+    ignore_unused(outputs);
+    return IsSupportedForDataTypeRef(reasonIfUnsupported,
+                                     input.GetDataType(),
+                                     &TrueFunc<>,
+                                     &TrueFunc<>);
+}
+
 bool RefLayerSupport::IsStridedSliceSupported(const TensorInfo& input,
                                               const TensorInfo& output,
                                               const StridedSliceDescriptor& descriptor,