Home
last modified time | relevance | path

Searched defs:reshapeOp (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/TransformOps/
H A DTensorTransformOps.cpp55 auto reshapeOp = cast<tensor::ReshapeOp>(op); in getNextOperands() local
65 auto reshapeOp = cast<ConcreteOp>(op); in getNextOperands() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp872 getBuffer(rewriter, reshapeOp.getSource(), options); in bufferize() local
856 auto reshapeOp = cast<tensor::ReshapeOp>(op); bufferizesToMemoryRead() local
910 auto reshapeOp = cast<tensor::ReshapeOp>(op); getBufferType() local
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp155 auto reshapeOp = cast<OpTy>(op); reifyResultShapes() local
H A DTensorOps.cpp1824 matchAndRewrite(TensorReshapeOp reshapeOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...]
/llvm-project/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h86 static OpFoldResult foldReshapeOp(ReshapeOpTy reshapeOp, in foldReshapeOp() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp776 fuseWithReshapeByExpansion(LinalgOp linalgOp,Operation * reshapeOp,OpOperand * fusableOpOperand,PatternRewriter & rewriter) fuseWithReshapeByExpansion() argument
934 tensor::CollapseShapeOp reshapeOp = matchAndRewrite() local
970 tensor::CollapseShapeOp reshapeOp = matchAndRewrite() local
1738 tensor::ExpandShapeOp reshapeOp = matchAndRewrite() local
1779 tensor::ExpandShapeOp reshapeOp = matchAndRewrite() local
[all...]
H A DTransforms.cpp336 auto reshapeOp = rewriter.create<tensor::ExpandShapeOp>( lowerPack() local
433 auto reshapeOp = rewriter.create<tensor::CollapseShapeOp>( lowerUnPack() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaCanonicalizations.cpp829 if (auto reshapeOp = llvm::dyn_cast_if_present<tosa::ReshapeOp>( REDUCE_FOLDER() local
/llvm-project/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1329 matchAndRewrite(ReshapeOp reshapeOp,typename ReshapeOp::Adaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
[all...]