Added a guard to prevent from crashing for certain cases.

PiperOrigin-RevId: 284272480 Change-Id: Idb6e566037d83406921e5d318aaa4d35a336edfa
frreiss · Dec 6, 2019 · bd04562 · bd04562
1 parent 1d96a61
commit bd04562
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/core/grappler/costs/graph_properties.cc b/tensorflow/core/grappler/costs/graph_properties.cc
@@ -1911,7 +1911,7 @@ Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner,
     // Infer the shape of the second output once and for all since it never
     // changes.
     ShapeHandle out1 = ic->Scalar();
-    ic->set_output(1, out1);
+    if (ic->num_outputs() >= 2) ic->set_output(1, out1);
   }
 
   ShapeHandle out;