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 @@
#include "ngraph/runtime/cpu/op/convert_layout.hpp"
#include "ngraph/runtime/cpu/cpu_layout_descriptor.hpp"
#include "ngraph/runtime/cpu/mkldnn_utils.hpp"
using namespace std;
using namespace ngraph;
......@@ -24,6 +25,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout(
const shared_ptr<Node>& arg, const shared_ptr<runtime::cpu::LayoutDescriptor>& layout)
: ConvertLayout(arg, 0, layout)
{
runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(this);
}
shared_ptr<Node>
......@@ -44,6 +46,7 @@ runtime::cpu::op::ConvertLayout::ConvertLayout(
, arg_output_index(output_index)
, output_layout(layout)
{
runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(this);
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