diff --git a/xla/array.cc b/xla/array.cc
index 84fbc12125dc8..b75941fbe87e7 100644
--- a/xla/array.cc
+++ b/xla/array.cc
@@ -16,6 +16,7 @@ limitations under the License.
 #include "xla/array.h"
 
 #include <complex>
+#include <cstdint>
 #include <random>
 
 #include "xla/types.h"
diff --git a/xla/array.h b/xla/array.h
index 0eab5a1b7360f..1d28388c56311 100644
--- a/xla/array.h
+++ b/xla/array.h
@@ -20,6 +20,7 @@ limitations under the License.
 #include <array>
 #include <cstdint>
 #include <cstring>
+#include <functional>
 #include <initializer_list>
 #include <iterator>
 #include <limits>
diff --git a/xla/array2d_test.cc b/xla/array2d_test.cc
index 4686e2ec5c1ac..921da30256fa3 100644
--- a/xla/array2d_test.cc
+++ b/xla/array2d_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/array2d.h"
 
+#include <cstdint>
 #include <initializer_list>
 #include <string>
 
diff --git a/xla/array3d_test.cc b/xla/array3d_test.cc
index 07599797fbbb0..334d733266b41 100644
--- a/xla/array3d_test.cc
+++ b/xla/array3d_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/array3d.h"
 
+#include <cstdint>
 #include <initializer_list>
 
 #include "xla/test.h"
diff --git a/xla/array4d_test.cc b/xla/array4d_test.cc
index 9acae92eef3b8..1deb1bc81f3c7 100644
--- a/xla/array4d_test.cc
+++ b/xla/array4d_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/array4d.h"
 
+#include <cstdint>
 #include <initializer_list>
 #include <numeric>
 #include <vector>
diff --git a/xla/array_test.cc b/xla/array_test.cc
index f8a8f09ba3629..bf79aa98f4049 100644
--- a/xla/array_test.cc
+++ b/xla/array_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/array.h"
 
+#include <cstdint>
 #include <initializer_list>
 #include <string>
 #include <vector>