Home
last modified time | relevance | path

Searched defs:unpackOp (Results 1 – 7 of 7) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Utils/
H A DUtils.cpp115 getUnPackInverseSrcPerm(UnPackOp unpackOp) getUnPackInverseSrcPerm() argument
121 getUnPackInverseSrcPerm(UnPackOp unpackOp,PackingMetadata & metadata) getUnPackInverseSrcPerm() argument
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp262 getUnpackTileDimInfo(OpBuilder & b,UnPackOp unpackOp,int64_t tileDim,OpFoldResult tileOffset,OpFoldResult tileSize) getUnpackTileDimInfo() argument
359 auto unpackOp = cast<UnPackOp>(op); getLoopIteratorTypes() local
387 auto unpackOp = cast<UnPackOp>(op); getTiledImplementation() local
[all...]
/llvm-project/mlir/lib/Conversion/GPUToNVVM/
H A DWmmaOpsToNvvm.cpp215 auto unpackOp = [&](Value operand) { in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DPackAndUnpackPatterns.cpp237 auto unpackOp = sliceOp.getSource().getDefiningOp<UnPackOp>(); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1567 vectorizeAsTensorUnpackOp(RewriterBase & rewriter,tensor::UnPackOp unpackOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorUnpackOp() argument
1789 vectorizeUnPackOpPrecondition(tensor::UnPackOp unpackOp,ArrayRef<int64_t> inputVectorSizes) vectorizeUnPackOpPrecondition() argument
1976 __anon729377362202(auto unpackOp) vectorizeOpPrecondition() argument
2074 __anon729377362702(auto unpackOp) vectorize() argument
[all...]
H A DDataLayoutPropagation.cpp1162 tensor::UnPackOp unpackOp = matchAndRewrite() local
H A DTransforms.cpp1209 matchAndRewrite(tensor::UnPackOp unpackOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...]