fix crash in ReshapeConvertLayout (#2205)

* fix crash in ngraph-tf test conv_ops_test.Conv2DTest.testConv2DKernelSmallerThanStrideSame * fix file perms * correct checks
NervanaSystems · Dec 11, 2018 · bf14a38 · bf14a38
1 parent 1640d21
commit bf14a38
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ngraph/runtime/cpu/pass/cpu_post_layout_optimizations.cpp b/src/ngraph/runtime/cpu/pass/cpu_post_layout_optimizations.cpp
@@ -212,7 +212,8 @@ void ngraph::runtime::cpu::pass::CPUPostLayoutOptimizations::
             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";
             return false;