Skip to content

Improvements for MKLDNN LRN#9329

Merged
luotao1 merged 1 commit intoPaddlePaddle:developfrom
tpatejko:tpatejko/mkldnn-lrn
Mar 27, 2018
Merged

Improvements for MKLDNN LRN#9329
luotao1 merged 1 commit intoPaddlePaddle:developfrom
tpatejko:tpatejko/mkldnn-lrn

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->SetOutputDim("MidOut", x_dim);
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

3 participants