Merge pull request #50061 from tensorflow/geetachavan1-patch-2

Fix CherryPick issue
tensorflow · Jun 4, 2021 · b2ef8a6 · b2ef8a6
2 parents c3405c3 + f9a1ba8
commit b2ef8a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow/core/kernels/sparse_fill_empty_rows_op.cc b/tensorflow/core/kernels/sparse_fill_empty_rows_op.cc
@@ -71,8 +71,7 @@ class SparseFillEmptyRowsOp : public OpKernel {
     // dense_shape.  Also add check that dense rank > 0.
     // Also add check that dense rank > 0.
     OP_REQUIRES(context, dense_shape_t.NumElements() != 0,
-                errors::InvalidArgument("Dense shape cannot be empty."),
-                done);
+                errors::InvalidArgument("Dense shape cannot be empty."));
     const T& default_value = default_value_t.scalar<T>()();
     const auto indices = indices_t.matrix<int64>();
     const auto values = values_t.vec<T>();