Commit 0f04d5ff authored by Diego Caballero's avatar Diego Caballero Committed by Scott Cyphers

[Standalone] Set MKLDNN kernel flag in ConvertLayout op. (#2697)

ConvertLayout is a MKLDNN specific op and the MKLDNN kernel flag was not
set. This led to assume that this operation was not supported by
MKLDNN.
parent 90edb4f6
...@@ -16,6 +16,7 @@ ...@@ -16,6 +16,7 @@
#include "ngraph/runtime/cpu/op/convert_layout.hpp" #include "ngraph/runtime/cpu/op/convert_layout.hpp"
#include "ngraph/runtime/cpu/cpu_layout_descriptor.hpp" #include "ngraph/runtime/cpu/cpu_layout_descriptor.hpp"
#include "ngraph/runtime/cpu/mkldnn_utils.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ngraph;
...@@ -24,6 +25,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout( ...@@ -24,6 +25,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout(
const shared_ptr<Node>& arg, const shared_ptr<runtime::cpu::LayoutDescriptor>& layout) const shared_ptr<Node>& arg, const shared_ptr<runtime::cpu::LayoutDescriptor>& layout)
: ConvertLayout(arg, 0, layout) : ConvertLayout(arg, 0, layout)
{ {
runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(this);
} }
shared_ptr<Node> shared_ptr<Node>
...@@ -44,6 +46,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout( ...@@ -44,6 +46,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout(
, arg_output_index(output_index) , arg_output_index(output_index)
, output_layout(layout) , output_layout(layout)
{ {
runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(this);
constructor_validate_and_infer_types(); constructor_validate_and_infer_types();
} }
......
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