Searched refs:bvm (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 450 BlockAndValueMapping bvm; in hoistPaddingOnTensors() local
456 if (bvm.lookupOrDefault(sliceOp.getSource()) == packedTensor) in hoistPaddingOnTensors()
461 b.clone(*op, bvm); in hoistPaddingOnTensors()
466 loc, bvm.lookupOrDefault(forOp.getLowerBound()), in hoistPaddingOnTensors()
467 bvm.lookupOrDefault(forOp.getUpperBound()), in hoistPaddingOnTensors()
468 bvm.lookupOrDefault(forOp.getStep()), packedTensor); in hoistPaddingOnTensors()
470 bvm.map(forOp.getInductionVar(), clonedForOp.getInductionVar()); in hoistPaddingOnTensors()
471 bvm.map(forOp.getRegionIterArgs(), clonedForOp.getRegionIterArgs()); in hoistPaddingOnTensors()
472 bvm.map(forOp.getResults(), clonedForOp.getResults()); in hoistPaddingOnTensors()
502 Value paddedTensor = bvm.lookup(opToHoist.getResult()); in hoistPaddingOnTensors()
[all …]
|
| H A D | Vectorization.cpp | 258 Value vectorValue = bvm.lookup(outputs.value()); in vectorizeLinalgYield()
308 Value reduceVec = bvm.lookup(reduceValue); in reduceIfNeeded()
309 Value outputVec = bvm.lookup(initialValue); in reduceIfNeeded()
342 const BlockAndValueMapping &bvm, in vectorizeOneOp() argument
349 VectorizationResult result = customFunc(op, bvm); in vectorizeOneOp()
383 reductionOperands[0].second, bvm); in vectorizeOneOp()
399 ? bvm.lookup(v) in vectorizeOneOp()
446 BlockAndValueMapping bvm; in vectorizeAsLinalgGeneric() local
465 bvm.map(bbarg, opOperand->get()); in vectorizeAsLinalgGeneric()
497 bvm.map(bbarg, readValue); in vectorizeAsLinalgGeneric()
[all …]
|
| H A D | Transforms.cpp | 745 BlockAndValueMapping bvm; in createFillOrGenerateOp() local
746 padOp.getRegion().cloneInto(&generateOp.getRegion(), bvm); in createFillOrGenerateOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 254 BlockAndValueMapping bvm; in bubbleUpPadSlice() local
255 padOp.getRegion().cloneInto(&newPadOp.getRegion(), bvm); in bubbleUpPadSlice()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 215 BlockAndValueMapping bvm; in outlineSingleBlockRegion() local
216 bvm.map(originalTerminator->getOperands(), call->getResults()); in outlineSingleBlockRegion()
217 rewriter.clone(*originalTerminator, bvm); in outlineSingleBlockRegion()
231 BlockAndValueMapping bvm; in outlineSingleBlockRegion() local
232 repl = rewriter.clone(*cst, bvm)->getResult(0); in outlineSingleBlockRegion()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.td | 1056 BlockAndValueMapping bvm;
1061 r.cloneInto(state.addRegion(), bvm);
1076 "ValueRange":$operands, "BlockAndValueMapping &":$bvm),
1082 r.cloneInto(state.addRegion(), bvm);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 924 BlockAndValueMapping bvm; in matchAndRewrite() local
925 bvm.map(contractionOp.getAcc(), otherOperand); in matchAndRewrite()
927 cast<vector::ContractionOp>(rewriter.clone(*contractionOp, bvm)); in matchAndRewrite()
|