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

Improvements for MKLDNN LRN #9329

Merged
merged 1 commit into from
Mar 27, 2018
Merged

Improvements for MKLDNN LRN #9329

merged 1 commit into from
Mar 27, 2018

Conversation

tpatejko
Copy link

@tpatejko tpatejko commented Mar 22, 2018

This PR builds on #9123 and adds the following:

  • refactoring and correcting how is_testattribute is handled by MKLDNN LRN's forward pass;
  • unit tests for is_test attribute

Copy link
Contributor

@tensor-tang tensor-tang left a comment

Choose a reason for hiding this comment

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

LGTM about mkldnn part.
@luotao1, could u please review the other parts.

ctx->ShareLoD("X", /*->*/ "Out");
ctx->SetOutputDim("MidOut", x_dim);
}
Copy link
Contributor

Choose a reason for hiding this comment

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

No need do this.

Copy link
Contributor

Choose a reason for hiding this comment

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

Yes, no need do this, but no other effect.

Copy link
Contributor

@luotao1 luotao1 left a comment

Choose a reason for hiding this comment

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

LGTM

@luotao1 luotao1 merged commit 1b67bc0 into PaddlePaddle:develop Mar 27, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants