From f624f597100c9e9351cf0dcc192db6c982783217 Mon Sep 17 00:00:00 2001
From: Yanan Cao <ycao@google.com>
Date: Fri, 30 Aug 2019 21:52:38 -0700
Subject: [PATCH] Fix a typo in ConvertToTensorShapeProto declaration.

PiperOrigin-RevId: 266509672
---
 tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h b/tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h
index 6e7b28e2a67..8c401090d42 100644
--- a/tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h
+++ b/tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h
@@ -42,7 +42,7 @@ StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
                                            mlir::Builder* builder);
 
 // Converts a shape from MLIR to an TensorFlow tensor shape proto.
-Status ConvertToTensorShapeProto(llvm::ArrayRef<int> shape,
+Status ConvertToTensorShapeProto(llvm::ArrayRef<int64_t> shape,
                                  TensorShapeProto* output_shape);
 
 // Converts an MLIR elements attribute to an TensorFlow tensor proto.