Update sparse_fill_empty_rows_op.cc

tensorflow · Jun 3, 2021 · f9a1ba8 · f9a1ba8
1 parent c3405c3
commit f9a1ba8
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>();