colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Restrict canonicalization to avoid changing type 

Issue #47516

PiperOrigin-RevId: 363300979
This commit is contained in:
Jacques Pienaar 2021-03-16 16:53:10 -07:00 committed by TensorFlow MLIR Team
parent caae2525ef
commit a58e62590e
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lib/Dialect/mhlo/IR/hlo_patterns.td
View File

@ -36,12 +36,13 @@ def DynamicBroadcastToOwnShape_4 : Pat<
(HLO_DynamicBroadcastInDimOp:$op $x, (Tensor_CastOp (Shape_ShapeOfOp $x)), $attr), (HLO_DynamicBroadcastInDimOp:$op $x, (Tensor_CastOp (Shape_ShapeOfOp $x)), $attr),
(Tensor_CastOp $x)>; (Tensor_CastOp $x)>;
def ShapeOfDynamicReshape : Pat<
(Shape_ShapeOfOp (HLO_DynamicReshapeOp $x, $shape)),
(replaceWithValue $shape)>;
def HasSameType : Constraint<CPred<"$0.getType() == $1.getType()">>; def HasSameType : Constraint<CPred<"$0.getType() == $1.getType()">>;
def ShapeOfDynamicReshape : Pat<
(Shape_ShapeOfOp:$op (HLO_DynamicReshapeOp $x, $shape)),
(replaceWithValue $shape),
[(HasSameType $shape, $op)]>;
def IdentityBroadcastReshape : Pat< def IdentityBroadcastReshape : Pat<
(HLO_ReshapeOp:$op (HLO_BroadcastOp $input, $dims)), (HLO_ReshapeOp:$op (HLO_BroadcastOp $input, $dims)),
(replaceWithValue $input), (replaceWithValue $input),