From 3826a0d08df35c8c936a1828420b8843ffd14cb0 Mon Sep 17 00:00:00 2001 From: Mateusz Bencer Date: Mon, 9 Aug 2021 06:26:37 +0200 Subject: [PATCH] Remove leaky relu alpha check (#6910) * remove leaky relu check * make alpha_node scalar --- ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp b/ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp index e40f85c4b6d..8b12626f0fa 100644 --- a/ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp +++ b/ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp @@ -21,11 +21,8 @@ namespace ngraph auto data = node.get_ng_inputs().at(0); double alpha = node.get_attribute_value("alpha", 0.01); - CHECK_VALID_NODE( - node, alpha >= 0 && alpha <= 1, " alpha value should be in range (0,1)"); - std::shared_ptr alpha_node = - default_opset::Constant::create(data.get_element_type(), Shape{}, {alpha}); + default_opset::Constant::create(data.get_element_type(), Shape{1}, {alpha}); return {std::make_shared(data, alpha_node)}; }