Merge pull request #54018 from pranve/cherrypick-1fb27733f943295d8744…

…17630edd3b38b34ce082-on-r2.7

Remove `CHECK`-fails from `IsSimplifiableReshape`

tensorflow/core/grappler/optimizers/constant_folding.cc

@@ -1689,7 +1689,11 @@ bool ConstantFolding::IsSimplifiableReshape(
   if (!IsReshape(node)) {
     return false;
   }
-  CHECK_LE(2, node.input_size());
+  if (2 > node.input_size()) {
+    return errors::Internal("Node ", node.name(),
+                            " must have at most 2 inputs but has ",
+                            node.input_size());
+  }
   const NodeDef* new_shape = node_map_->GetNode(node.input(1));
   if (!IsReallyConstant(*new_shape)) {
     return false;
@@ -1705,7 +1709,11 @@ bool ConstantFolding::IsSimplifiableReshape(
   if (!s.ok()) {
     return false;
   }
-  CHECK_EQ(1, outputs.size());
+  if (outputs.size() != 1) {
+    return errors::Internal("Node ", node.name(),
+                            " must have exactly 1 output but has ",
+                            outputs.size());
+  }
 
   const std::vector<OpInfo::TensorProperties>& props =
       properties.GetInputProperties(node.name());