Fix a typo in ConvertToTensorShapeProto declaration.
PiperOrigin-RevId: 266509672
This commit is contained in:
parent
98bd5a6e22
commit
f624f59710
@ -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.
|
||||
|
Loading…
x
Reference in New Issue
Block a user