-
Notifications
You must be signed in to change notification settings - Fork 2.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ONNX] GRU and RNN operators. #607
Conversation
- Update few tests accuracy tolerance - Update rnn_fwd_activations with new reference values and model.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great objective abstraction for recurrent onnx ops!
{ | ||
const auto& ng_inputs = node.get_ng_inputs(); | ||
|
||
m_map[OpInput::X] = ng_inputs.at(0); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we add assert to ensure that all inputs have static shape?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good point. Maybe not all are required, I'll review it once again and try to assert only what's necessary.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, but please resolve the comments and conflicts
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, but OpenVINO don't support W/R/B as inputs, please add several tests where W/R/B are Constants.
@itikhono I've just added one such test case, and although I don't get errors about transformation, IE_CPU produces bad results. |
:( We need to find the root cause. I'll investigate it today. |
The IE_CPU test with W,R as constants will fail due to the lack of support for multioutput in nG IE backend. Nevertheless the first output provides correct results. |
No description provided.