saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use forward_inference to speed up convolutions with MKLDNN (#731)

Browse source
This commit is contained in:
Mika Fischer 2019-03-29 18:42:50 +01:00 committed by Changming Sun
parent e6a2bdfacd
commit 7f96c7f028
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/mkldnn/nn/conv.cc
View file

@ -171,13 +171,13 @@ class ConvPrimitive : public PrimitiveBase {
if (!params.bias_dims.empty()) {
context_.fwd_desc.reset(new mkldnn::convolution_forward::desc(
mkldnn::prop_kind::forward, mkldnn::convolution_direct, *context_.src_md,
mkldnn::prop_kind::forward_inference, mkldnn::convolution_direct, *context_.src_md,
*context_.filter_md, *context_.bias_md, *context_.dst_md,
params.strides, params.dilations, params.padding_left,
params.padding_right, mkldnn::padding_kind::zero));
} else {
context_.fwd_desc.reset(new mkldnn::convolution_forward::desc(
mkldnn::prop_kind::forward, mkldnn::convolution_direct, *context_.src_md,
mkldnn::prop_kind::forward_inference, mkldnn::convolution_direct, *context_.src_md,
*context_.filter_md, *context_.dst_md, params.strides,
params.dilations, params.padding_left,
params.padding_right, mkldnn::padding_kind::zero));