diff --git a/lib/Dialect/mhlo/IR/hlo_patterns.td b/lib/Dialect/mhlo/IR/hlo_patterns.td index b8b6cb8..bdb3e3c 100644 --- a/lib/Dialect/mhlo/IR/hlo_patterns.td +++ b/lib/Dialect/mhlo/IR/hlo_patterns.td @@ -18,15 +18,13 @@ limitations under the License. include "mlir/Dialect/Shape/IR/ShapeOps.td" include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.td" -def EqualBinaryOperands : Constraint>; - // Canonicalization patterns. def DynamicBroadcastToOwnShape_1 : Pat< - (HLO_DynamicBroadcastInDimOp:$op $arg0, - (Shape_ToExtentTensorOp (Shape_ShapeOfOp $arg1)), $attr), - (replaceWithValue $arg0), [(EqualBinaryOperands $arg0, $arg1)]>; + (HLO_DynamicBroadcastInDimOp:$op $x, + (Shape_ToExtentTensorOp (Shape_ShapeOfOp $x)), $attr), + (replaceWithValue $x)>; def DynamicBroadcastToOwnShape_2 : Pat< - (HLO_DynamicBroadcastInDimOp:$op $arg0, (Shape_ShapeOfOp $arg1), $attr), - (replaceWithValue $arg0), [(EqualBinaryOperands $arg0, $arg1)]>; + (HLO_DynamicBroadcastInDimOp:$op $x, (Shape_ShapeOfOp $x), $attr), + (replaceWithValue $x)>;