diff options
author | 2016-11-17 15:37:00 -0800 | |
---|---|---|
committer | 2016-11-17 15:45:18 -0800 | |
commit | 54e5000e0b980abe905900599c4493fadae34a15 (patch) | |
tree | c1f0751e5565882a77646589360fe090bec3f3fc /tensorflow/core/kernels/gather_nd_op_gpu.cu.cc | |
parent | 8a5610cd9f0b7087c1a7e97071ba1cf9b885315a (diff) |
Merge changes from github.
Change: 139516555
Diffstat (limited to 'tensorflow/core/kernels/gather_nd_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/gather_nd_op_gpu.cu.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/gather_nd_op_gpu.cu.cc b/tensorflow/core/kernels/gather_nd_op_gpu.cu.cc index dd9f83afbc..56ffe58569 100644 --- a/tensorflow/core/kernels/gather_nd_op_gpu.cu.cc +++ b/tensorflow/core/kernels/gather_nd_op_gpu.cu.cc @@ -72,8 +72,8 @@ struct GatherNdSlice<GPUDevice, T, Index, IXDIM> { Eigen::array<int64, IXDIM> batch_strides; Eigen::array<int64, IXDIM> batch_indices; if (IXDIM > 0) { - batch_strides[IXDIM - 1] = s_size; - batch_indices[IXDIM - 1] = Tparams.dimension(IXDIM - 1); + batch_strides[size_t(IXDIM - 1)] = s_size; + batch_indices[size_t(IXDIM - 1)] = Tparams.dimension(IXDIM - 1); } for (int i = IXDIM - 1; i > 0; --i) { batch_indices[i - 1] = Tparams.dimension(i - 1); |