From d54eb295a6bd0287bfd2faefe90acb72a4f3c17c Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Thu, 28 Mar 2019 18:13:22 -0700
Subject: [PATCH] [XLA] Revert mistaken comment change.

PiperOrigin-RevId: 240894481
---
 tensorflow/compiler/xla/shape_util.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/compiler/xla/shape_util.h b/tensorflow/compiler/xla/shape_util.h
index b13cac5db84..ee24e39f052 100644
--- a/tensorflow/compiler/xla/shape_util.h
+++ b/tensorflow/compiler/xla/shape_util.h
@@ -624,7 +624,7 @@ class ShapeUtil {
   // continue, or false otherwise.
   //
   // visitor_function must be a callable of type
-  // StatusOr<bool>(ShapeIndex) or compatible.
+  // StatusOr<bool>(absl::Span<int64>) or compatible.
   template <typename FnType>
   static Status ForEachIndexWithStatus(const Shape& shape,
                                        absl::Span<const int64> base,