Home
last modified time | relevance | path

Searched refs:collapseOp (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h230 LogicalResult matchAndRewrite(CollapseOpTy collapseOp, in matchAndRewrite()
237 ShapedType resultType = collapseOp.getResultType(); in matchAndRewrite()
239 if (hasNonIdentityLayout(collapseOp.getSrc().getType()) || in matchAndRewrite()
255 lowerRankReassociation = collapseOp.getReassociationIndices(); in matchAndRewrite()
257 higherRankReassociation = collapseOp.getReassociationIndices(); in matchAndRewrite()
292 if (!collapseOp) in matchAndRewrite()
295 ShapedType srcType = collapseOp.getSrcType(); in matchAndRewrite()
299 hasNonIdentityLayout(collapseOp.getSrc().getType()) || in matchAndRewrite()
300 hasNonIdentityLayout(collapseOp.getResult().getType())) in matchAndRewrite()
308 auto srcReassociation = collapseOp.getReassociationIndices(); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp129 if (auto collapseOp = in runOnOperation() local
131 if (!collapseOp.getSrc().getDefiningOp<linalg::LinalgOp>()) { in runOnOperation()