Skip to content
Permalink
Browse files Browse the repository at this point in the history
Remove CHECK-fails from IsSimplifiableReshape
PiperOrigin-RevId: 409164987
Change-Id: I58c7dd459ff348c3dbae95e00c4c5e63b30a4e65
  • Loading branch information
mihaimaruseac authored and tensorflower-gardener committed Nov 11, 2021
1 parent ebc1a2f commit 1fb2773
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions tensorflow/core/grappler/optimizers/constant_folding.cc
Expand Up @@ -1689,7 +1689,11 @@ Status ConstantFolding::IsSimplifiableReshape(
if (!IsReshape(node)) {
return errors::Internal("Node ", node.name(), " is not a Reshape node");
}
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 errors::Internal("Node ", node.name(), " has shape ",
Expand All @@ -1707,7 +1711,11 @@ Status ConstantFolding::IsSimplifiableReshape(
if (!s.ok()) {
return errors::Internal("Could not evaluate node ", node.name());
}
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());
Expand Down

0 comments on commit 1fb2773

Please sign in to comment.