diff --git a/tensorflow/compiler/xla/mlir_hlo/mhlo/IR/hlo_ops.cc b/tensorflow/compiler/xla/mlir_hlo/mhlo/IR/hlo_ops.cc index f13fd274b38..43fdd57dbe9 100644 --- a/tensorflow/compiler/xla/mlir_hlo/mhlo/IR/hlo_ops.cc +++ b/tensorflow/compiler/xla/mlir_hlo/mhlo/IR/hlo_ops.cc @@ -1541,9 +1541,10 @@ LogicalResult DynamicIotaOp::reifyReturnTypeShapes( //===----------------------------------------------------------------------===// // DynamicUpdateSliceOp //===----------------------------------------------------------------------===// + LogicalResult DynamicUpdateSliceOp::inferReturnTypes( MLIRContext*, std::optional location, ValueRange operands, - DictionaryAttr attributes, RegionRange regions, + DictionaryAttr attributes, OpaqueProperties, RegionRange regions, SmallVectorImpl& inferredReturnTypes) { DynamicUpdateSliceOp::Adaptor adaptor(operands, attributes, {}, regions); RankedTensorType operandType =