ELU

inference-engine/tests/functional/plugin/shared/src/single_layer_tests/activation.cpp

@@ -169,7 +169,6 @@ void ActivationParamLayerTest::Infer() {
     inferRequest.Infer();
 }
 
-
 void ActivationParamLayerTest::SetUp() {
     InferenceEngine::Precision netPrecision;
     std::pair<std::vector<size_t>, std::vector<size_t>> shapes;

ngraph/core/include/ngraph/op/elu.hpp

@@ -48,6 +48,9 @@ namespace ngraph
                     clone_with_new_inputs(const OutputVector& new_args) const override;
 
                 double get_alpha() const { return m_alpha; }
+
+                bool evaluate(const HostTensorVector& outputs,
+                              const HostTensorVector& inputs) const override;
             private:
                 double m_alpha;
             };

ngraph/core/include/ngraph/runtime/reference/elu.hpp

@@ -30,7 +30,7 @@ namespace ngraph
             {
                 for (size_t i = 0; i < count; i++)
                 {
-                    out[i] = arg[i] < 0 ? alpha * (std::exp(arg[i]) - 1.0) : arg[i];
+                    out[i] = arg[i] < T(0) ? T(alpha * (std::exp(arg[i]) - 1.0)) : arg[i];
                 }
             }
         }

ngraph/core/src/op/elu.cpp

@@ -14,6 +14,7 @@
 // limitations under the License.
 //*****************************************************************************
 
+#include <ngraph/runtime/reference/elu.hpp>
 #include "ngraph/op/elu.hpp"
 #include "ngraph/attribute_visitor.hpp"
 #include "ngraph/builder/autobroadcast.hpp"
@@ -48,3 +49,37 @@ shared_ptr<Node> op::Elu::clone_with_new_inputs(const OutputVector& new_args) co
     check_new_args_count(this, new_args);
     return make_shared<Elu>(new_args.at(0), m_alpha);
 }
+
+
+template <element::Type_t ET>
+bool evaluate(const HostTensorPtr& arg,const HostTensorPtr& out, double alpha)
+{
+    runtime::reference::elu(arg->get_data_ptr<ET>(),
+                                out->get_data_ptr<ET>(),
+                                shape_size(arg->get_shape()),
+                                alpha);
+    return true;
+}
+
+bool evaluate_elu(const HostTensorPtr& arg, const HostTensorPtr& out, double alpha)
+{
+    bool rc = true;
+    switch (arg->get_element_type())
+    {
+        TYPE_CASE(i8)(arg, out, alpha);
+            break;
+        TYPE_CASE(bf16)(arg, out, alpha);
+            break;
+        TYPE_CASE(f16)(arg, out, alpha);
+            break;
+        TYPE_CASE(f32)(arg, out, alpha);
+            break;
+        default: rc = false; break;
+    }
+    return rc;
+}
+
+
+bool op::v0::Elu::evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const {
+    return evaluate_elu(inputs[0], outputs[0], m_alpha);
+}