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

Update Convolution layer/op weight layout to not depend on input format #6412

merged 15 commits into from Sep 12, 2018


Copy link

commented Sep 11, 2018

Fixes: #6393

ConvolutionUtils<T>::mkldnn_conv2d(*block.getMKLDNNStream(), {input, weights, bias}, output, {kH,kW,sH,sW,pH,pW,dH,dW,isSameMode,isNCHW});
} else {

This comment has been minimized.

Copy link

saudet Sep 11, 2018


Please don't do that, MKL-DNN does support strides:

@AlexDBlack AlexDBlack merged commit 45c84e1 into master Sep 12, 2018

0 of 2 checks passed

codeclimate Code Climate encountered an error attempting to analyze this pull request.
continuous-integration/jenkins/pr-head This commit cannot be built

@AlexDBlack AlexDBlack deleted the shyrma_weights_format branch Sep 12, 2018

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
None yet
2 participants
You can’t perform that action at this time.