OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update sparse_fill_empty_rows_op.cc

Browse Source
This commit is contained in:
geetachavan1
2021-06-03 16:33:08 -07:00
committed by GitHub
parent afba115566
commit 26e6e9e6e2
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/kernels/sparse_fill_empty_rows_op.cc
View File

@@ -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>();