Skip to content

N
ngraph
Commit 1cede467 authored by gaurides's avatar gaurides Committed by Scott Cyphers
Browse files
Options

Fixed MKLDNN convolution prop_kind (#2813)

parent 2c361e80
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
src/ngraph/runtime/cpu/mkldnn_emitter.cpp
View file @ 1cede467
......@@ -280,7 +280,7 @@ mkldnn::memory::format MKLDNNEmitter::query_convolution_forward_weight_format(
mkldnn::algorithm convolution_algo = mkldnn_utils::get_conv_algo();
mkldnn::engine cpu_engine(mkldnn::engine::cpu, 0);
mkldnn::convolution_forward::desc conv_desc_layout(
mkldnn::prop_kind::forward,
mkldnn::prop_kind::forward_inference,
convolution_algo,
input_data_desc,
weights_desc_any, // this needs to be in default format
......@@ -397,7 +397,7 @@ size_t MKLDNNEmitter::build_convolution_forward(const mkldnn::memory::desc& inpu
try
{
auto conv_prim = new mkldnn::convolution_forward(
{{mkldnn::prop_kind::forward,
{{mkldnn::prop_kind::forward_inference,
convolution_algo,
input_data_desc,
weights_desc,
......@@ -547,7 +547,7 @@ size_t MKLDNNEmitter::build_convolution_forward(const mkldnn::memory::desc& inpu
try
{
conv_index = insert_primitive(new mkldnn::convolution_forward(
{{mkldnn::prop_kind::forward,
{{mkldnn::prop_kind::forward_inference,
convolution_algo,
input_data_desc,
weights_desc,
......
src/ngraph/runtime/cpu/mkldnn_emitter.hpp
View file @ 1cede467
......@@ -1364,7 +1364,7 @@ namespace ngraph
{
auto bias_desc = mkldnn_utils::get_input_mkldnn_md(node, 2);
return mkldnn::convolution_forward::desc(
mkldnn::prop_kind::forward,
mkldnn::prop_kind::forward_inference,
convolution_algo,
data_desc,
weights_desc,
......@@ -1379,7 +1379,7 @@ namespace ngraph
else
{
return mkldnn::convolution_forward::desc(
mkldnn::prop_kind::forward,
mkldnn::prop_kind::forward_inference,
convolution_algo,
data_desc,
weights_desc,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment