Skip to content
This repository was archived by the owner on Oct 13, 2021. It is now read-only.

Refactor RNN parameter extracting #405

Merged
merged 6 commits into from
Mar 9, 2020

Conversation

cjermain
Copy link
Contributor

@cjermain cjermain commented Mar 8, 2020

This PR is a step towards supporting SimpleRNN and GRU in the Bidirectional layer, by moving toward a consistent API across the RNN layers. It refactors the RNN code to increase sharing, and re-uses that in the Bidirectional LSTM case. No functional changes are made -- this PR is purely a refactor.

@wenbingl, can you review these changes?

@wenbingl
Copy link
Member

wenbingl commented Mar 9, 2020

Thanks a lot for making RNN converter better!

@wenbingl wenbingl merged commit 7544e13 into onnx:master Mar 9, 2020
@cjermain cjermain deleted the feature/rnn_refactor branch March 10, 2020 00:08
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants