9df6afbb5c
Lower `shape.any` to its first operand. Differential Revision: https://reviews.llvm.org/D83123
21 lines
721 B
TableGen
21 lines
721 B
TableGen
include "mlir/Dialect/Shape/IR/ShapeOps.td"
|
|
include "mlir/Dialect/StandardOps/IR/Ops.td"
|
|
|
|
// Convert `from_extent_tensor` and `to_extent_tensor` to no-ops as shapes will
|
|
// be represented as extent tensors.
|
|
def FromExtentTensorOpConversion : Pat<
|
|
(Shape_FromExtentTensorOp $input),
|
|
(replaceWithValue $input)>;
|
|
def ToExtentTensorOpConversion : Pat<
|
|
(Shape_ToExtentTensorOp $input),
|
|
(replaceWithValue $input)>;
|
|
|
|
// Convert `index_to_size` and `size_to_index` to no-ops as sizes will be
|
|
// represented as indices.
|
|
def IndexToSizeOpConversion : Pat<
|
|
(Shape_IndexToSizeOp $arg),
|
|
(replaceWithValue $arg)>;
|
|
def SizeToIndexOpConversion : Pat<
|
|
(Shape_SizeToIndexOp $arg),
|
|
(replaceWithValue $arg)>;
|