Skip to content

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.

test_forward_split_squeeze()
test_forward_expand_dims()
test_forward_pooling()
test_forward_lrn() No newline at end of file
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.

4 participants