diff --git a/tensorflow/compiler/xla/service/hlo_evaluator.cc b/tensorflow/compiler/xla/service/hlo_evaluator.cc
index 72b215217f4c51..a0c5cbe916050a 100644
--- a/tensorflow/compiler/xla/service/hlo_evaluator.cc
+++ b/tensorflow/compiler/xla/service/hlo_evaluator.cc
@@ -646,8 +646,9 @@ class HloEvaluator::TypedVisitor : public DfsHloVisitorWithDefault {
                                  0);
     std::vector<int64> step(evaluated_operand.shape().dimensions_size(), 1);
 
-    ShapeUtil::ForEachIndex(evaluated_operand.shape(), zero_base,
-                            evaluated_operand.shape().dimensions(), step, func);
+    ShapeUtil::ForEachIndex(
+        evaluated_operand.shape(), zero_base,
+        AsInt64Slice(evaluated_operand.shape().dimensions()), step, func);
 
     parent_->evaluated_[pad] = std::move(result);
     return Status::OK();