Home
last modified time | relevance | path

Searched defs:collapseOp (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DExtractSliceFromReshapeUtils.cpp64 OpBuilder &b, tensor::CollapseShapeOp collapseOp, in create()
/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp194 if (auto collapseOp = dyn_cast<tensor::CollapseShapeOp>(producer)) { in runOnOperation() local
/llvm-project/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h327 auto collapseOp in matchAndRewrite() local
[all...]
/llvm-project/mlir/test/lib/Dialect/Tensor/
H A DTestTensorTransforms.cpp167 auto collapseOp = op.getSource().getDefiningOp<tensor::CollapseShapeOp>(); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DDataLayoutPropagation.cpp651 bubbleUpPackOpThroughCollapseShape(tensor::CollapseShapeOp collapseOp, in bubbleUpPackOpThroughCollapseShape() argument
H A DElementwiseOpFusion.cpp1077 if (auto collapseOp = matchAndRewrite() local