diff options
Diffstat (limited to 'tensorflow/core/ops/control_flow_ops.cc')
| -rw-r--r-- | tensorflow/core/ops/control_flow_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/ops/control_flow_ops.cc b/tensorflow/core/ops/control_flow_ops.cc index 0a7c704b0a..645ae43553 100644 --- a/tensorflow/core/ops/control_flow_ops.cc +++ b/tensorflow/core/ops/control_flow_ops.cc @@ -122,7 +122,7 @@ Status MergeShape(InferenceContext* c) { int32 rank = c->Rank(out); for (int i = 1; i < c->num_inputs(); ++i) { ShapeHandle input = c->input(i); - if (c->Rank(input) != rank) { + if (!c->RankKnown(input) || c->Rank(input) != rank) { out = c->UnknownShape(); break; } |