Commit 5a1e3c5c authored by Tomasz Dołbniak's avatar Tomasz Dołbniak Committed by Scott Cyphers

Use v1 ops in ONNX ThresholdedRelu (#4096)

Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
parent 0f2734dc
...@@ -18,9 +18,6 @@ ...@@ -18,9 +18,6 @@
#include <vector> #include <vector>
#include "default_opset.hpp" #include "default_opset.hpp"
#include "ngraph/op/multiply.hpp"
#include "ngraph/op/util/broadcasting.hpp"
#include "ngraph/opsets/opset0.hpp"
#include "thresholded_relu.hpp" #include "thresholded_relu.hpp"
namespace ngraph namespace ngraph
...@@ -33,18 +30,17 @@ namespace ngraph ...@@ -33,18 +30,17 @@ namespace ngraph
{ {
NodeVector thresholded_relu(const Node& node) NodeVector thresholded_relu(const Node& node)
{ {
auto data = node.get_ng_inputs().at(0); const auto data = node.get_ng_inputs().at(0);
double alpha = node.get_attribute_value<double>("alpha", 1.0); const double alpha = node.get_attribute_value<double>("alpha", 1.0);
std::shared_ptr<ngraph::Node> alpha_node = const auto alpha_node = default_opset::Constant::create(
std::make_shared<default_opset::Constant>(data->get_element_type(), data->get_element_type(), data->get_shape(), {alpha});
data->get_shape(),
std::vector<double>{alpha});
auto data_map = std::make_shared<default_opset::Convert>( const auto data_map = std::make_shared<default_opset::Convert>(
std::make_shared<ngraph::opset0::Greater>(data, alpha_node), std::make_shared<default_opset::Greater>(data, alpha_node),
data->get_element_type()); data->get_element_type());
return {data * data_map};
return {std::make_shared<default_opset::Multiply>(data, data_map)};
} }
} // namespace set_1default_opset } // namespace set_1default_opset
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment