| /llvm-project/flang/lib/Optimizer/Transforms/ |
| H A D | AbstractResult.cpp | 153 llvm::SmallVector<mlir::Value> newOperands; in matchAndRewrite()
155 newOperands.emplace_back(arg); in matchAndRewrite()
156 newOperands.append(op.getOperands().begin(), op.getOperands().end()); in matchAndRewrite()
158 newResultTypes, newOperands); in matchAndRewrite()
169 llvm::SmallVector<mlir::Value> newOperands; in matchAndRewrite()
170 newOperands.push_back( in matchAndRewrite()
173 newOperands.push_back(arg); in matchAndRewrite()
174 newOperands.append(op.getOperands().begin() + 1, in matchAndRewrite()
177 newResultTypes, newOperands); in matchAndRewrite()
183 llvm::SmallVector<mlir::Value> newOperands; in matchAndRewrite()
124 llvm::SmallVector<mlir::Value> newOperands; matchAndRewrite() local
140 llvm::SmallVector<mlir::Value> newOperands; matchAndRewrite() local
154 llvm::SmallVector<mlir::Value> newOperands; matchAndRewrite() local
[all...] |
| H A D | ConstantArgumentGlobalisation.cpp | 47 llvm::SmallVector<mlir::Value> newOperands; in matchAndRewrite() local
55 newOperands.push_back(a); in matchAndRewrite()
80 newOperands.push_back(a); in matchAndRewrite()
90 newOperands.push_back(a); in matchAndRewrite()
119 newOperands.push_back(addr); in matchAndRewrite()
132 newResultTypes, newOperands); in matchAndRewrite()
|
| /llvm-project/mlir/lib/Dialect/Func/Transforms/ |
| H A D | DecomposeCallGraphTypes.cpp | |
| H A D | FuncConversions.cpp | 93 SmallVector<Value, 4> newOperands(op->operand_begin(), op->operand_end());
105 newOperands[idx] = operands[idx];
109 op, [newOperands, op]() { op->setOperands(newOperands); }); in matchAndRewrite()
72 SmallVector<Value, 4> newOperands(op->operand_begin(), op->operand_end()); matchAndRewrite() local
|
| /llvm-project/mlir/include/mlir/Dialect/Utils/ |
| H A D | StructuredOpsUtils.h | 143 ValueRange newOperands);
146 ValueRange newOperands) { in clone() argument
147 return cast<OpT>(clone(b, op.getOperation(), newResultTypes, newOperands)); in clone()
154 ValueRange newOperands);
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | InlineScalarOperands.cpp | 43 SmallVector<Value> newOperands; in matchAndRewrite() local
50 newOperands.emplace_back(opOperand->get()); in matchAndRewrite()
64 loc, genericOp->getResultTypes(), newOperands, outputOperands, in matchAndRewrite()
|
| H A D | Padding.cpp | 182 SmallVector<Value> newOperands; in rewriteAsPaddedOp() local
183 newOperands.reserve(opToPad->getNumOperands()); in rewriteAsPaddedOp()
194 newOperands.push_back(*paddedOperand); in rewriteAsPaddedOp()
210 ValueRange(newOperands).take_back(opToPad.getNumDpsInits()).getTypes(); in rewriteAsPaddedOp()
212 paddedOp = clone(rewriter, opToPad, resultTensorTypes, newOperands); in rewriteAsPaddedOp()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 70 SmallVector<Value> newOperands = newInputBuffers; in bufferizeDestinationStyleOpInterface() local
71 newOperands.append(newOutputBuffers.begin(), newOutputBuffers.end()); in bufferizeDestinationStyleOpInterface()
79 OperationState state(op->getLoc(), op->getName(), newOperands, TypeRange{}, in bufferizeDestinationStyleOpInterface()
|
| H A D | MeshShardingInterfaceImpl.cpp | 184 SmallVector<Value> newOperands = llvm::to_vector(spmdizedOperands); in createDestinationPassingStyleInitOperands()
188 newOperands[operandIdx] = createDestinationPassingStyleInitOperand( in createAllReduceForResultWithoutPartialSharding()
190 return newOperands; in createAllReduceForResultWithoutPartialSharding()
177 SmallVector<Value> newOperands = llvm::to_vector(spmdizedOperands); createDestinationPassingStyleInitOperands() local
|
| H A D | DropUnitDims.cpp | 499 SmallVector<Value> newOperands; in dropUnitDims()
503 newOperands.push_back(opOperand.get()); in dropUnitDims()
506 newOperands.push_back(collapseValue(rewriter, loc, opOperand.get(), in dropUnitDims()
514 ArrayRef<Value>(newOperands).take_front(genericOp.getNumDpsInputs()); in dropUnitDims()
516 ArrayRef<Value>(newOperands).take_back(genericOp.getNumDpsInits()); in dropUnitDims()
500 SmallVector<Value> newOperands; dropUnitDims() local
|
| H A D | EraseUnusedOperandsAndResults.cpp | 259 SmallVector<OpOperand *> &newOperands, in populateOpPayload()
265 OpOperand *newOperand = newOperands[it->second]; in populateOpPayload()
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorDropLeadUnitDim.cpp | 368 SmallVector<Value> newOperands; in castAwayContractionLeadingOneDim() local
447 newOperands.push_back( in castAwayContractionLeadingOneDim()
456 loc, newOperands[0], newOperands[1], newOperands[2], in castAwayContractionLeadingOneDim()
520 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
523 newOperands.push_back(rewriter.create<vector::ExtractOp>( in matchAndRewrite()
526 newOperands.push_back(operand); in matchAndRewrite()
531 newOperands, newVecType, op->getAttrs()); in matchAndRewrite()
|
| H A D | VectorDistribute.cpp | 626 SmallVector<Value> newOperands(elementWise->getOperands().begin(), in matchAndRewrite()
629 newOperands[i] = newWarpOp.getResult(newRetIndices[i]); in matchAndRewrite()
634 rewriter, loc, elementWise, newOperands, in matchAndRewrite()
1058 SmallVector<Value> newOperands;
1068 newOperands.push_back(maskDimIdx); in matchAndRewrite()
1072 rewriter.create<vector::CreateMaskOp>(loc, distType, newOperands); in matchAndRewrite()
1583 SmallVector<Value> newOperands; in matchAndRewrite()
1590 newOperands.push_back( in matchAndRewrite()
1603 forOp.getStep(), newOperands); in matchAndRewrite()
714 SmallVector<Value> newOperands(elementWise->getOperands().begin(), matchAndRewrite() local
1197 SmallVector<Value> newOperands; matchAndRewrite() local
1702 SmallVector<Value> newOperands; matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Utils/ |
| H A D | StructuredOpsUtils.cpp | 198 ValueRange newOperands) { in clone() argument
200 OperationState state(op->getLoc(), op->getName(), newOperands, newResultTypes, in clone()
211 ValueRange newOperands) { in cloneWithoutRegions() argument
212 OperationState state(op->getLoc(), op->getName(), newOperands, newResultTypes, in cloneWithoutRegions()
|
| /llvm-project/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | BufferResultsToOutParams.cpp | 199 auto newOperands = llvm::to_vector<6>(op.getOperands()); in updateCalls()
200 newOperands.append(outParams.begin(), outParams.end()); in updateCalls()
204 newResultTypes, newOperands); in promoteBufferResultsToOutParams()
189 auto newOperands = llvm::to_vector<6>(op.getOperands()); updateCalls() local
|
| H A D | OwnershipBasedBufferDeallocation.cpp | 811 SmallVector<Value> newOperands(forwardedOperands); in handleInterface() local
814 newOperands.append(additionalConditions.begin(), additionalConditions.end()); in handleInterface()
815 op.getSuccessorOperands(0).getMutableForwardedOperands().assign(newOperands); in handleInterface()
966 SmallVector<Value> newOperands{operands.getAsOperandRange()}; in handleInterface() local
967 newOperands.append(updatedOwnerships.begin(), updatedOwnerships.end()); in handleInterface()
968 operands.assign(newOperands); in handleInterface()
|
| H A D | FuncBufferizableOpInterfaceImpl.cpp | 245 SmallVector<Value> newOperands; in bufferize()
253 newOperands.push_back(opOperand.get()); in bufferize()
294 newOperands.push_back(buffer); in bufferizesToMemoryRead()
299 callOp.getLoc(), funcOp.getSymName(), resultTypes, newOperands); in bufferizesToMemoryWrite()
242 SmallVector<Value> newOperands; bufferize() local
|
| H A D | Bufferize.cpp | 482 SmallVector<Value> newOperands; in bufferizeOp()
487 newOperands.push_back(operand); in bufferizeOp()
502 newOperands.push_back(bufferizedOperand); in bufferizeOp()
504 operands.getMutableForwardedOperands().assign(newOperands); in bufferizeOp()
595 SmallVector<Value> newOperands; bufferizeBlockSignature() local
|
| /llvm-project/mlir/lib/IR/ |
| H A D | OperationSupport.cpp | 207 void OperationState::addOperands(ValueRange newOperands) { in setProperties()
208 operands.append(newOperands.begin(), newOperands.end()); in setProperties()
357 MutableArrayRef<OpOperand> newOperands(newOperandStorage, newSize); in resize()
359 newOperands.begin()); in resize()
367 new (&newOperands[numOperands]) OpOperand(owner); in resize()
377 return newOperands; in resize()
212 addOperands(ValueRange newOperands) addOperands() argument
362 MutableArrayRef<OpOperand> newOperands(newOperandStorage, newSize); resize() local
|
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopFusionUtils.cpp | 367 SmallVector<Value> newOperands; in promoteSingleIterReductionLoop() local
368 llvm::append_range(newOperands, in promoteSingleIterReductionLoop()
378 return newOperands; in promoteSingleIterReductionLoop()
|
| /llvm-project/mlir/lib/Conversion/VectorToSPIRV/ |
| H A D | VectorToSPIRV.cpp | 580 SmallVector<Value> newOperands(mask.size());
581 for (auto [shuffleIdx, newOperand] : llvm::zip_equal(mask, newOperands)) {
593 if (newOperands.size() == 1) { in matchAndRewrite()
594 rewriter.replaceOp(shuffleOp, newOperands.front()); in matchAndRewrite()
599 shuffleOp, newResultType, newOperands); in matchAndRewrite()
626 Value newOperands[] = {adaptor.getLhs(), adaptor.getRhs()}; in matchAndRewrite()
628 interleaveOp, newResultType, newOperands); in matchAndRewrite()
560 SmallVector<Value> newOperands(mask.size()); matchAndRewrite() local
601 Value newOperands[] = {adaptor.getLhs(), adaptor.getRhs()}; matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 2423 SmallVector<Value> newOperands = linalgOp.getDpsInputs(); in matchAndRewrite()
2427 newOperands.append(outputOperands.begin(), outputOperands.end()); in matchAndRewrite()
2432 Operation *newOp = clone(rewriter, linalgOp, resultTypes, newOperands); in matchAndRewrite()
2484 /// mapped in `affineExprToSize`. New operands are created in `newOperands` and
2487 /// `newOperands` list. in getTiledImplementation()
2491 SmallVector<Value> &newOperands, SmallVector<Type> &resultTypes, in getTiledImplementation()
2494 newOperands.push_back(src); in getTiledImplementation()
2529 newOperands[index] = newOperand; in reifyResultShapes()
2560 SmallVector<Value> newOperands; in getEffects()
2566 newOperands in getEffects()
2260 SmallVector<Value> newOperands = linalgOp.getDpsInputs(); matchAndRewrite() local
2328 createNewOperandWithStaticSizes(Location loc,PatternRewriter & rewriter,OpOperand * opOperand,llvm::DenseMap<AffineExpr,int64_t> & affineExprToSize,LinalgOp linalgOp,SmallVector<Value> & newOperands,SmallVector<Type> & resultTypes,bool & changeNeeded) createNewOperandWithStaticSizes() argument
2397 SmallVector<Value> newOperands; matchAndRewrite() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 2150 SmallVector<Value> newOperands; in simplifyConstrainedMinMaxOp() local
2151 unpackOptionalValues(constraints.getMaybeValues(), newOperands); in simplifyConstrainedMinMaxOp()
2156 if (!newOperands[i] || getConstantIntValue(newOperands[i])) in simplifyConstrainedMinMaxOp()
2167 affine::canonicalizeMapAndOperands(&newMap, &newOperands); in simplifyConstrainedMinMaxOp()
2168 return AffineValueMap(newMap, newOperands); in simplifyConstrainedMinMaxOp()
|
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 3413 auto newOperands =
3415 rewriter.replaceOpWithNewOp<T>(affineOp, newMap, newOperands);
3840 SmallVector<Value, 4> newOperands(lbOperands); in setUpperBounds()
3841 newOperands.append(ubOperands.begin(), ubOperands.end()); in setUpperBounds()
3842 (*this)->setOperands(newOperands); in setUpperBounds() local
3851 SmallVector<Value, 4> newOperands(getLowerBoundsOperands()); in setSteps()
3852 newOperands.append(ubOperands.begin(), ubOperands.end());
3853 (*this)->setOperands(newOperands);
3961 SmallVector<Value, 4> newOperands{operands};
3963 composeAffineMapAndOperands(&newMap, &newOperands); in canonicalizeLoopBounds()
3404 auto newOperands = matchAndRewrite() local
3831 SmallVector<Value, 4> newOperands(lbOperands); setLowerBounds() local
3952 SmallVector<Value, 4> newOperands{operands}; canonicalize() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Shape/IR/ |
| H A D | Shape.cpp | 698 auto newOperands = llvm::filter_to_vector<8>(op->getOperands(), in matchAndRewrite() local
703 if (newOperands.empty()) { in matchAndRewrite()
710 if (newOperands.size() < op.getNumOperands()) {
711 rewriter.replaceOpWithNewOp<OpTy>(op, op->getResultTypes(), newOperands,
808 auto newOperands = llvm::to_vector<8>( in matchAndRewrite()
814 rewriter.replaceOpWithNewOp<OpTy>(op, op->getResultTypes(), newOperands);
800 auto newOperands = llvm::to_vector<8>( matchAndRewrite() local
|