[nnc] Fix incorrect operation id for leaky relu (#2815)

Id of LeakyRelu operation was incorrectly set as ReLU

Signed-off-by: Vladimir Plazun <v.plazun@partner.samsung.com>

diff --git a/contrib/nnc/include/core/modelIR/operations/LeakyReluOp.h b/contrib/nnc/include/core/modelIR/operations/LeakyReluOp.h
index b968a07..0ff8964 100644 (file)
--- a/contrib/nnc/include/core/modelIR/operations/LeakyReluOp.h
+++ b/contrib/nnc/include/core/modelIR/operations/LeakyReluOp.h
@@ -26,7 +26,7 @@ namespace ops {
 class LeakyReluOp : public Operation {
 public:
   explicit LeakyReluOp(const IODescriptor& arg, float alpha)
-  : Operation(Type::ReLU, {arg}), _alpha(alpha) {
+  : Operation(Type::leakyReLU, {arg}), _alpha(alpha) {
     // Infer output shape.
     setOutputShape(0, getInputShape(0));
   }

diff --git a/contrib/nnc/unittests/soft_backend/CPPOperations.cpp b/contrib/nnc/unittests/soft_backend/CPPOperations.cpp
index da93b33..e3068cb 100644 (file)
--- a/contrib/nnc/unittests/soft_backend/CPPOperations.cpp
+++ b/contrib/nnc/unittests/soft_backend/CPPOperations.cpp
@@ -810,7 +810,7 @@ TEST(cpp_operations_test, leaky_relu) {
     return g.create<mir::ops::LeakyReluOp>("y", inputs[0], 0.1);
   };
 
-  createAndRunTestGraph(op_generator, relu, input_ntensors, input_atensor);
+  createAndRunTestGraph(op_generator, leakyRelu, input_ntensors, input_atensor);
 }
 
 TEST(cpp_operations_test, sigmoid) {