Searched refs:collapsedType (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 104 T collapsedType, bool isExpansion) { in verifyReshapeLikeTypes() argument
106 unsigned collapsedRank = collapsedType.getRank(); in verifyReshapeLikeTypes()
110 << " to have higher rank than the type = " << collapsedType; in verifyReshapeLikeTypes()
139 return verifyReshapeLikeShapes(op, collapsedType, expandedType, isExpansion); in verifyReshapeLikeTypes()
156 static LogicalResult verifyReshapeLikeShapes(OpTy op, ShapedType collapsedType, in verifyReshapeLikeShapes() argument
161 collapsedType.getShape(), expandedType.getShape(), in verifyReshapeLikeShapes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 866 RankedTensorType collapsedType) { in verifyTensorReshapeOp() argument
868 verifyReshapeLikeTypes(op, expandedType, collapsedType, isExpansion))) in verifyTensorReshapeOp()
874 if (!isSameTypesWithoutEncoding(collapsedType, expectedType)) in verifyTensorReshapeOp()
876 << expectedType << ", but got " << collapsedType; in verifyTensorReshapeOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseOpFusion.cpp | 696 RankedTensorType collapsedType = isExpanding in fuseWithReshapeByExpansion() local
705 expandedType.getShape(), collapsedType.getShape(), rewriter))) in fuseWithReshapeByExpansion()
|