Update hard_sigmoid.cpp (#4)

luo-cheng2021 · Jul 23, 2021 · 23efd25 · 23efd25
1 parent a17505b
commit 23efd25
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/ngraph/frontend/paddlepaddle/src/op/hard_sigmoid.cpp b/ngraph/frontend/paddlepaddle/src/op/hard_sigmoid.cpp
@@ -3,7 +3,6 @@
 //
 
 #include "hard_sigmoid.hpp"
-#include <ngraph/builder/make_constant.hpp>
 #include <ngraph/opsets/opset6.hpp>
 #include <paddlepaddle_frontend/utility.hpp>
 
@@ -21,8 +20,8 @@ namespace ngraph
                     auto dtype = data.get_element_type();
                     float slope = node.get_attribute<float>("slope", 0.2f);
                     float offset = node.get_attribute<float>("offset", 0.5f);
-                    auto alpha = builder::make_constant(dtype, {}, slope);
-                    auto beta = builder::make_constant(dtype, {}, offset);
+                    auto alpha = ngraph::opset6::Constant::create(dtype, Shape{}, {slope});
+                    auto beta = ngraph::opset6::Constant::create(dtype, Shape{}, {offset});
                     return node.default_single_output_mapping(
                         {std::make_shared<ngraph::opset6::HardSigmoid>(data, alpha, beta)},
                         {"Out"});
@@ -31,4 +30,4 @@ namespace ngraph
             } // namespace op
         }     // namespace pdpd
     }         // namespace frontend
-} // namespace ngraph
+} // namespace ngraph