Skip to content
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

[MXNET] LRN support in MXNET frontend #1520

Merged
merged 2 commits into from
Aug 2, 2018

Conversation

PariksheetPinjari909
Copy link
Contributor

@PariksheetPinjari909 PariksheetPinjari909 commented Aug 1, 2018

This PR adds LRN support in MXNET frontend

@kazum @srkreddy1238 pls review.

Copy link
Contributor

@srkreddy1238 srkreddy1238 left a comment

Choose a reason for hiding this comment

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

With minor change LGTM.

new_attrs['alpha'] = attrs.get('alpha', 0.0001)
new_attrs['beta'] = attrs.get('beta', 0.75)
new_attrs['bias'] = attrs.get('knorm', 2)
new_attrs['axis'] = attrs.get('axis', 1)
Copy link
Contributor

Choose a reason for hiding this comment

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

Could directly assign axis as mxnet never give this attribute.
Also suggest give a hint about why axis is 1.

Copy link
Contributor

@kazum kazum left a comment

Choose a reason for hiding this comment

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

Added a minor comment. Other parts look good to me.

@@ -163,3 +168,4 @@ def test_forward_pooling():
test_forward_split_squeeze()
test_forward_expand_dims()
test_forward_pooling()
test_forward_lrn()
Copy link
Contributor

Choose a reason for hiding this comment

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

Add a new line at the end of file.

@PariksheetPinjari909
Copy link
Contributor Author

@kazum @srkreddy1238 Review comments are handled now. Pls check.

Copy link
Contributor

@srkreddy1238 srkreddy1238 left a comment

Choose a reason for hiding this comment

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

Thanks @PariksheetPinjari909 .

LGTM.

Copy link
Contributor

@kazum kazum left a comment

Choose a reason for hiding this comment

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

Looks good to me, thanks!

@tqchen tqchen merged commit 54ca149 into apache:master Aug 2, 2018
@tqchen
Copy link
Member

tqchen commented Aug 2, 2018

Thanks @PariksheetPinjari909 @kazum @srkreddy1238 , this is merged

tqchen pushed a commit to tqchen/tvm that referenced this pull request Aug 4, 2018
sergei-mironov pushed a commit to sergei-mironov/tvm that referenced this pull request Aug 8, 2018
@PariksheetPinjari909 PariksheetPinjari909 deleted the mxnet_lrn branch August 14, 2018 09:32
Sign up for free to 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.

None yet

4 participants