Skip to content

N
ngraph
Commit 7bad261f authored by Jayaram Bobba's avatar Jayaram Bobba
Browse files
Options

Style fixes

parent bbe8da24
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
src/ngraph/runtime/cpu/pass/cpu_layout.cpp
View file @ 7bad261f
...@@ -525,9 +525,9 @@ namespace ngraph ...@@ -525,9 +525,9 @@ namespace ngraph
{ {
// TODO (jbobba): Check with MKLDNN folks if this is necessary // TODO (jbobba): Check with MKLDNN folks if this is necessary
throw ngraph_error("MKLDNN Unsupported pooling layout" + throw ngraph_error("MKLDNN Unsupported pooling layout" +
to_string(input_layout)); to_string(input_layout) + e.message);
prim_input_formats.push_back(memory::format::nchw); // prim_input_formats.push_back(memory::format::nchw);
prim_output_formats.push_back(memory::format::nchw); // prim_output_formats.push_back(memory::format::nchw);
} }
node = node =
...@@ -612,9 +612,9 @@ namespace ngraph ...@@ -612,9 +612,9 @@ namespace ngraph
{ {
// TODO (jbobba): Check with MKLDNN folks if this is necessary // TODO (jbobba): Check with MKLDNN folks if this is necessary
throw ngraph_error("MKLDNN Unsupported pooling layout" + throw ngraph_error("MKLDNN Unsupported pooling layout" +
to_string(input_layout)); to_string(input_layout) + e.message);
prim_input_formats.push_back(memory::format::nchw); // prim_input_formats.push_back(memory::format::nchw);
prim_output_formats.push_back(memory::format::nchw); // prim_output_formats.push_back(memory::format::nchw);
} }
node = node =
...@@ -651,7 +651,7 @@ namespace ngraph ...@@ -651,7 +651,7 @@ namespace ngraph
{ {
auto input_layout = auto input_layout =
runtime::cpu::mkldnn_utils::get_input_mkldnn_format(node.get(), 0); runtime::cpu::mkldnn_utils::get_input_mkldnn_format(node.get(), 0);
vector<memory::format> prim_input_formats; vector<memory::format> prim_input_formats;
vector<memory::format> prim_output_formats; vector<memory::format> prim_output_formats;
prim_input_formats.push_back(input_layout); prim_input_formats.push_back(input_layout);
......
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