Fix a typo in ConvertToTensorShapeProto declaration.

PiperOrigin-RevId: 266509672
This commit is contained in:
Yanan Cao 2019-08-30 21:52:38 -07:00 committed by TensorFlower Gardener
parent 98bd5a6e22
commit f624f59710

View File

@ -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.