Skip to content

[FRONTEND][DARKNET]LSTM and GRU support - #1576

Merged
tqchen merged 3 commits into
apache:masterfrom
siju-samuel:darknet_lstm_gru
Aug 14, 2018
Merged

[FRONTEND][DARKNET]LSTM and GRU support#1576
tqchen merged 3 commits into
apache:masterfrom
siju-samuel:darknet_lstm_gru

Conversation

@siju-samuel

Copy link
Copy Markdown
Member

All the RNN operators are completed for Darknet

Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from others in the community.

@siju-samuel

Copy link
Copy Markdown
MemberAuthor

@PariksheetPinjari909@srkreddy1238@yzhliu could you please review this PR.

@PariksheetPinjari909PariksheetPinjari909 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Others looks LGTM

Comment threadnnvm/python/nnvm/frontend/darknet.py Outdated

elif LAYERTYPE.GRU == layer.type:
if layer.steps > 1:
raise NotImplementedError("Currenty support only single step GRU")

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

pls correct spell.

Comment threadnnvm/python/nnvm/frontend/darknet.py Outdated

elif LAYERTYPE.LSTM == layer.type:
if layer.steps > 1:
raise NotImplementedError("Currenty support only single step GRU")

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

pls correct spell

test_forward_crnn()
test_forward_lstm()
test_forward_gru()
test_forward_activation_logistic() No newline at end of file

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add end line

@siju-samuel

Copy link
Copy Markdown
MemberAuthor

@PariksheetPinjari909 Thanks for the review. I updated the code. Please check again. TIA.

@tqchen
tqchen merged commit 54a115e into apache:masterAug 14, 2018
@tqchen

Copy link
Copy Markdown
Member

Thanks @siju-samuel@PariksheetPinjari909 , this is now merged

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@siju-samuel@tqchen@PariksheetPinjari909@yzhliu