commit e662a940d3378cfe669ff7e259a6911713fc0df9
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Oct 14 15:12:00 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Tue Oct 15 11:46:03 2019 +0000
tree 40c6b83f2d4b936da053bcb74784a8a849c4c9bc
parent f4d59a678d8ef8420f52d341bb60f1a583269e24

IVGCVSW-3975 Add reference workload for LOG_SOFTMAX

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I10bb7133e0e2d6d7199abdf39562b1226bbbd3e7

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 0d6b16c..9342b29 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -897,6 +897,32 @@
     return supported;
 }
 
+bool RefLayerSupport::IsLogSoftmaxSupported(const TensorInfo& input,
+                                            const TensorInfo& output,
+                                            const LogSoftmaxDescriptor& descriptor,
+                                            Optional<std::string&> reasonIfUnsupported) const
+{
+    ignore_unused(descriptor);
+
+    std::array<DataType, 2> supportedTypes =
+    {
+            DataType::Float32,
+            DataType::Float16
+    };
+
+    bool supported = true;
+    supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported,
+                                  "Reference LogSoftmax: input type not supported");
+
+    supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported,
+                                  "Reference LogSoftmax: output type not supported");
+
+    supported &= CheckSupportRule(TypesAreEqual(input, output), reasonIfUnsupported,
+                                  "Reference LogSoftmax: input and output types do not match");
+
+    return supported;
+}
+
 bool RefLayerSupport::IsLstmSupported(const TensorInfo& input,
                                       const TensorInfo& outputStateIn,
                                       const TensorInfo& cellStateIn,
@@ -1499,13 +1525,13 @@
     };
 
     supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported,
-                                  "Reference concatenation: output type not supported");
+                                  "Reference Softmax: output type not supported");
 
     supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported,
-                                  "Reference concatenation: input type not supported");
+                                  "Reference Softmax: input type not supported");
 
     supported &= CheckSupportRule(TypesAreEqual(input, output), reasonIfUnsupported,
-                                  "Reference concatenation: input type not supported");
+                                  "Reference Softmax: input type not supported");
 
     return supported;
 }