Commit 6584306c authored by gaurides's avatar gaurides Committed by Scott Cyphers

fix crash in ReshapeConvertLayout (#2205)

* fix crash in ngraph-tf test conv_ops_test.Conv2DTest.testConv2DKernelSmallerThanStrideSame

* fix file perms

* correct checks
parent 24bd105f
...@@ -212,7 +212,8 @@ void ngraph::runtime::cpu::pass::CPUPostLayoutOptimizations:: ...@@ -212,7 +212,8 @@ void ngraph::runtime::cpu::pass::CPUPostLayoutOptimizations::
return false; return false;
} }
if (reshape_m->get_op_annotations()->get_in_place_oi_pairs().size() == 0) auto annotation = reshape_m->get_op_annotations();
if (!annotation || annotation->get_in_place_oi_pairs().size() == 0)
{ {
NGRAPH_DEBUG << "ReshapeConvertLayout: Reshape is not pass-through"; NGRAPH_DEBUG << "ReshapeConvertLayout: Reshape is not pass-through";
return false; return false;
......
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