diff options
Diffstat (limited to 'tensorflow/core/kernels/topk_op.cc')
-rw-r--r-- | tensorflow/core/kernels/topk_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/topk_op.cc b/tensorflow/core/kernels/topk_op.cc index 630fcb76f3..5c89eaef5f 100644 --- a/tensorflow/core/kernels/topk_op.cc +++ b/tensorflow/core/kernels/topk_op.cc @@ -93,7 +93,7 @@ class TopK : public OpKernel { rows_by_one.set(0, num_rows); #else Eigen::array<int, 1> reduce_on_cols = {1}; - Eigen::array<int, 1> rows_by_one = {static_cast<int>(num_rows), 1}; + Eigen::array<int, 2> rows_by_one = {static_cast<int>(num_rows), 1}; #endif values.device(d) = |