Merge pull request #50064 from tensorflow/geetachavan1-patch-5

Fix Cherrypick issue
This commit is contained in:
Mihai Maruseac 2021-06-03 17:10:45 -07:00 committed by GitHub
commit 6ab3dc891b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -67,8 +67,7 @@ class SparseFillEmptyRowsOp : public OpKernel {
// TODO(ebrevdo): add shape checks between values, indices,
// dense_shape. 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>();