diff --git a/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter.cc b/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter.cc
index 59dbcbf0600..aa28a36c945 100644
--- a/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter.cc
+++ b/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter.cc
@@ -58,9 +58,10 @@ struct ShapeInfo {
   mlir::Type element_type;
 };
 
-ShapeInfo GetShapeInfo(const Shape& shape, int64 n_dim, int64 c_dim,
-                       absl::Span<const int64> spatial_dims,
-                       mlir::Builder builder) {
+ShapeInfo GetShapeInfo(
+    const Shape& shape, int64 n_dim, int64 c_dim,
+    absl::Span<const tensorflow::protobuf_int64> spatial_dims,
+    mlir::Builder builder) {
   ShapeInfo shape_info;
 
   std::vector<int64> physical_to_logical(