Commit ad6d1d43 authored by Amy Zhuang's avatar Amy Zhuang Committed by Scott Cyphers

Fix a bug in layout pass. (#4145)

parent d755e693
......@@ -1796,14 +1796,8 @@ namespace ngraph
}
catch (const mkldnn::error& e)
{
#if MKLDNN_VERSION_MAJOR < 1
throw ngraph_error("MKLDNN Unsupported pooling layout" +
to_string(fprop_input_md.data.format) +
MKLDNN_ERROR_MESSAGE);
#else
throw ngraph_error("MKLDNN Unsupported pooling layout" +
MKLDNN_ERROR_MESSAGE);
#endif
}
}
......
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