Skip to content

N
ngraph
Commit 6716068c authored by gaurides's avatar gaurides Committed by Robert Kimball
Browse files
Options

Fix accuracy issue (#2652) 

* Fix accuracy issue

* Style fix
parent 29dd3e3f
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 0 deletions
src/ngraph/runtime/cpu/pass/cpu_layout.cpp
View file @ 6716068c
......@@ -1362,6 +1362,8 @@ namespace ngraph
template <>
void CPULayout::LAYOUT_DECL(ngraph::op::Quantize)
{
if (mkldnn_utils::use_mkldnn_kernel(node.get()))
{
auto input_md = mkldnn_utils::get_input_mkldnn_md(node.get(), 0);
auto tv = node->get_output_tensor_ptr(0);
......@@ -1387,9 +1389,16 @@ namespace ngraph
set_output_layouts(node, o_mds);
}
}
else
{
set_native_layouts(external_function, node);
}
}
template <>
void CPULayout::LAYOUT_DECL(ngraph::op::Dequantize)
{
if (mkldnn_utils::use_mkldnn_kernel(node.get()))
{
auto input_md = mkldnn_utils::get_input_mkldnn_md(node.get(), 0);
auto tv = node->get_output_tensor_ptr(0);
......@@ -1413,6 +1422,11 @@ namespace ngraph
set_output_layouts(node, o_mds);
}
}
else
{
set_native_layouts(external_function, node);
}
}
template <>
void CPULayout::LAYOUT_DECL(ngraph::op::MaxPoolWithIndices)
......
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