diff options
author | 2016-07-22 09:40:27 -0800 | |
---|---|---|
committer | 2016-07-22 10:47:36 -0700 | |
commit | 68f3e2e6f38de4c9c6763a7db5510bd695413a82 (patch) | |
tree | 00d08ecc8a4d35147707fc8203da246cb91b4330 /tensorflow/core/kernels/one_hot_op.cc | |
parent | 3758878728710801f681afba39e145df4ddb8bf1 (diff) |
OneHotOp should use int for prefix_dim_size not index type.
Change: 128187997
Diffstat (limited to 'tensorflow/core/kernels/one_hot_op.cc')
-rw-r--r-- | tensorflow/core/kernels/one_hot_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/one_hot_op.cc b/tensorflow/core/kernels/one_hot_op.cc index 126e7f1cb8..916d85df2b 100644 --- a/tensorflow/core/kernels/one_hot_op.cc +++ b/tensorflow/core/kernels/one_hot_op.cc @@ -88,7 +88,7 @@ class OneHotOp : public OpKernel { // prefix_dim_size == # of elements before the axis // depth_v == # of elements per axis // suffix_dim_size == # of elements after the axis - TI prefix_dim_size = 1; + int64 prefix_dim_size = 1; for (int i = 0; i < axis; ++i) { prefix_dim_size *= indices_shape.dim_size(i); } |