From 7f96c7f0281a853fc9627dc050445b76fd3e0377 Mon Sep 17 00:00:00 2001 From: Mika Fischer Date: Fri, 29 Mar 2019 18:42:50 +0100 Subject: [PATCH] Use forward_inference to speed up convolutions with MKLDNN (#731) --- onnxruntime/core/providers/mkldnn/nn/conv.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/mkldnn/nn/conv.cc b/onnxruntime/core/providers/mkldnn/nn/conv.cc index e5184992e9..ab1827092d 100644 --- a/onnxruntime/core/providers/mkldnn/nn/conv.cc +++ b/onnxruntime/core/providers/mkldnn/nn/conv.cc @@ -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));