Unverified Commit 8e180988 authored by Michał Karzyński's avatar Michał Karzyński Committed by GitHub

Change ONNX LeakyReLU to use PReLU op (#4359)

Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
parent e1d9ee66
......@@ -19,8 +19,6 @@
#include "default_opset.hpp"
#include "exceptions.hpp"
#include "leaky_relu.hpp"
#include "ngraph/op/constant.hpp"
#include "ngraph/op/multiply.hpp"
namespace ngraph
{
......@@ -42,7 +40,7 @@ namespace ngraph
std::make_shared<default_opset::Constant>(data->get_element_type(),
data->get_shape(),
std::vector<double>{alpha});
return {std::make_shared<default_opset::Maximum>(data * alpha_node, data)};
return {std::make_shared<default_opset::PRelu>(data, alpha_node)};
}
} // namespace set_1
......
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