diff options
| author |  Patrick Nguyen <drpng@google.com> | 2017-01-12 18:09:40 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-12 18:33:38 -0800 |
| commit | 769496bb0d3093a5531268635d364950c93327d6 (patch) | |
| tree | 52c41f118733eecdbcd470a5cf12c37100a1e7e0 /tensorflow/core/kernels/sparse_split_op.cc | |
| parent | bb4d61d06a47d9783e4f3b66161fcfc084ab2273 (diff) | |
Merge changes from github.
Change: 144396000
Diffstat (limited to 'tensorflow/core/kernels/sparse_split_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/sparse_split_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/sparse_split_op.cc b/tensorflow/core/kernels/sparse_split_op.cc index 2ea986cf7d..9f8ed5ab18 100644 --- a/tensorflow/core/kernels/sparse_split_op.cc +++ b/tensorflow/core/kernels/sparse_split_op.cc @@ -30,7 +30,7 @@ class SparseSplitOp : public OpKernel { } void Compute(OpKernelContext* context) override { - const int32 split_dim = context->input(0).scalar<int>()(); + const int64 split_dim = context->input(0).scalar<int64>()(); const Tensor& input_indices = context->input(1); const Tensor& input_values = context->input(2); const Tensor& input_shape = context->input(3); |