IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove leaky relu alpha check (#6910)

Browse Source 
* remove leaky relu check

* make alpha_node scalar
This commit is contained in:
Mateusz Bencer 2021-08-09 06:26:37 +02:00 committed by GitHub
parent b11a2220b0
commit 3826a0d08d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ngraph/frontend/onnx/onnx_import/src/op/leaky_relu.cpp
View File

@ -21,11 +21,8 @@ namespace ngraph
auto data = node.get_ng_inputs().at(0);
double alpha = node.get_attribute_value<double>("alpha", 0.01);
CHECK_VALID_NODE(
node, alpha >= 0 && alpha <= 1, " alpha value should be in range (0,1)");
std::shared_ptr<ngraph::Node> 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<default_opset::PRelu>(data, alpha_node)};
}