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

DL4J: MKLDNN batch norm backprop support #8172

Closed
AlexDBlack opened this issue Aug 29, 2019 · 2 comments · Fixed by KonduitAI/deeplearning4j#43
Closed

DL4J: MKLDNN batch norm backprop support #8172

AlexDBlack opened this issue Aug 29, 2019 · 2 comments · Fixed by KonduitAI/deeplearning4j#43
Labels
Milestone

Comments

@AlexDBlack
Copy link
Contributor

@AlexDBlack AlexDBlack commented Aug 29, 2019

Currently DL4J supports MKL-DNN for batch norm forward pass only. Backprop needs to be added - requires this first: #8171

@shyrma

This comment has been minimized.

Copy link
Contributor

@shyrma shyrma commented Oct 26, 2019

Hi, Alex
MKLDNN api support for batch norm backprop has been provided and merged

@AlexDBlack AlexDBlack added this to the 1.0.0-beta6 milestone Oct 31, 2019
@AlexDBlack

This comment has been minimized.

Copy link
Contributor Author

@AlexDBlack AlexDBlack commented Nov 1, 2019

Blocked on: #8335

AlexDBlack added a commit to KonduitAI/deeplearning4j that referenced this issue Nov 13, 2019
Signed-off-by: AlexDBlack <blacka101@gmail.com>
AlexDBlack added a commit to KonduitAI/deeplearning4j that referenced this issue Nov 14, 2019
Signed-off-by: AlexDBlack <blacka101@gmail.com>
AlexDBlack added a commit to KonduitAI/deeplearning4j that referenced this issue Nov 14, 2019
* eclipse#8172 Enable DL4J MKLDNN batch norm backward pass

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* eclipse#8382 INDArray.toString() rank 1 brackets / ambiguity fix

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* eclipse#8308 Fix handful of broken links (inc. some in errors)

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Unused dependencies, round 1

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Unused dependencies, round 2

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Unused dependencies, round 3

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Small fix

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Uniform distribution TF import fix

Signed-off-by: AlexDBlack <blacka101@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
2 participants
You can’t perform that action at this time.