experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #25702 from yongtang:25701-dim_size-int

Browse Source 
PiperOrigin-RevId: 236189226
This commit is contained in:
TensorFlower Gardener 2019-02-28 15:43:20 -08:00
commit f929a5dbd9
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/sparse_dense_binary_op_shared.cc
View File

@ -78,9 +78,6 @@ class SparseDenseBinaryOpShared : public OpKernel {
"but received shapes: ", "but received shapes: ",
values_t->shape().DebugString(), " and ", values_t->shape().DebugString(), " and ",
shape_t->shape().DebugString())); shape_t->shape().DebugString()));
OP_REQUIRES(ctx, indices_t->dim_size(0) < std::numeric_limits<int>::max(),
errors::InvalidArgument(
"Number of non-zero elements exceeds int32 range"));
const auto indices_mat = indices_t->matrix<int64>(); const auto indices_mat = indices_t->matrix<int64>();
const auto shape_vec = shape_t->vec<int64>(); const auto shape_vec = shape_t->vec<int64>();
@ -106,7 +103,7 @@ class SparseDenseBinaryOpShared : public OpKernel {
Tensor *output_values = nullptr; Tensor *output_values = nullptr;
Tensor dense_gathered; Tensor dense_gathered;
const int nnz = static_cast<int>(indices_t->dim_size(0)); const int64 nnz = indices_t->dim_size(0);
OP_REQUIRES_OK(ctx, OP_REQUIRES_OK(ctx,
ctx->allocate_output(0, TensorShape({nnz}), &output_values)); ctx->allocate_output(0, TensorShape({nnz}), &output_values));
OP_REQUIRES_OK( OP_REQUIRES_OK(