Skip to content

[MXNET] LRN support in MXNET frontend#1520

Merged
tqchen merged 2 commits into
apache:masterfrom
PariksheetPinjari909:mxnet_lrn
Aug 2, 2018
Merged

[MXNET] LRN support in MXNET frontend#1520
tqchen merged 2 commits into
apache:masterfrom
PariksheetPinjari909:mxnet_lrn

Conversation

@PariksheetPinjari909

@PariksheetPinjari909 PariksheetPinjari909 commented Aug 1, 2018

Copy link
Copy Markdown
Contributor

This PR adds LRN support in MXNET frontend

@kazum @srkreddy1238 pls review.

@srkreddy1238 srkreddy1238 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.

With minor change LGTM.

Comment thread nnvm/python/nnvm/frontend/mxnet.py Outdated
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
Copy Markdown
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.

@kazum kazum 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.

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

test_forward_split_squeeze()
test_forward_expand_dims()
test_forward_pooling()
test_forward_lrn() 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 a new line at the end of file.

@PariksheetPinjari909

Copy link
Copy Markdown
Contributor Author

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

@srkreddy1238 srkreddy1238 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.

Thanks @PariksheetPinjari909 .

LGTM.

@kazum kazum 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.

Looks good to me, thanks!

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

tqchen commented Aug 2, 2018

Copy link
Copy Markdown
Member

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.

4 participants