Skip to content

N
ngraph
Commit 78d6a880 authored by Jayaram Bobba's avatar Jayaram Bobba Committed by Scott Cyphers
Browse files
Options

Minor change to proposal op validation and change to codeowners (#3046)

parent ea166bcd
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
CODEOWNERS
View file @ 78d6a880
...@@ -38,6 +38,7 @@ ...@@ -38,6 +38,7 @@
/src/ngraph/frontend/ @postrational /src/ngraph/frontend/ @postrational
/src/ngraph/op/ @diyessi @aprocter /src/ngraph/op/ @diyessi @aprocter
/src/ngraph/op/allreduce.*pp @wenzhe-nrv @diyessi @aprocter /src/ngraph/op/allreduce.*pp @wenzhe-nrv @diyessi @aprocter
/src/ngraph/op/experimental/layers @ilyachur
/src/ngraph/pass/ @jbobba /src/ngraph/pass/ @jbobba
/src/ngraph/pattern/ @jbobba @aprocter /src/ngraph/pattern/ @jbobba @aprocter
/src/ngraph/runtime/ @rkimballn1 @jbobba /src/ngraph/runtime/ @rkimballn1 @jbobba
......
src/ngraph/op/experimental/layers/proposal.cpp
View file @ 78d6a880
...@@ -45,8 +45,8 @@ void op::Proposal::validate_and_infer_types() ...@@ -45,8 +45,8 @@ void op::Proposal::validate_and_infer_types()
if (auto const_shape = dynamic_pointer_cast<op::Constant>(get_argument(2))) if (auto const_shape = dynamic_pointer_cast<op::Constant>(get_argument(2)))
{ {
NODE_VALIDATION_CHECK(this, NODE_VALIDATION_CHECK(this,
shape_size(const_shape->get_shape()) == 2, shape_size(const_shape->get_shape()) >= 1,
"Layer shape must have rank 2", "Layer shape must have rank greater than 1",
const_shape->get_shape()); const_shape->get_shape());
auto image_shape = const_shape->get_shape_val(); auto image_shape = const_shape->get_shape_val();
......
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