Home
last modified time | relevance | path

Searched refs:newOperands (Results 1 – 24 of 24) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Func/Transforms/
H A DDecomposeCallGraphTypes.cpp112 SmallVector<Value, 2> newOperands; in matchAndRewrite() local
115 operand, newOperands); in matchAndRewrite()
116 rewriter.replaceOpWithNewOp<ReturnOp>(op, newOperands); in matchAndRewrite()
139 SmallVector<Value, 2> newOperands; in matchAndRewrite() local
142 operand, newOperands); in matchAndRewrite()
161 newResultTypes, newOperands); in matchAndRewrite()
H A DFuncConversions.cpp67 SmallVector<Value, 4> newOperands(op->operand_begin(), op->operand_end()); in matchAndRewrite() local
79 newOperands[idx] = operands[idx]; in matchAndRewrite()
83 op, [newOperands, op]() { op->setOperands(newOperands); }); in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAbstractResult.cpp91 llvm::SmallVector<mlir::Value> newOperands = {arg}; in matchAndRewrite() local
92 newOperands.append(callOp.getOperands().begin(), in matchAndRewrite()
95 newOperands); in matchAndRewrite()
104 llvm::SmallVector<mlir::Value> newOperands; in matchAndRewrite() local
105 newOperands.push_back( in matchAndRewrite()
107 newOperands.push_back(arg); in matchAndRewrite()
108 newOperands.append(callOp.getOperands().begin() + 1, in matchAndRewrite()
111 newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DAffineCanonicalizationUtils.cpp184 SmallVector<Value> newOperands; in canonicalizeMinMaxOp() local
185 unpackOptionalValues(constraints.getMaybeValues(), newOperands); in canonicalizeMinMaxOp()
190 if (!newOperands[i] || getConstantIntValue(newOperands[i])) in canonicalizeMinMaxOp()
193 newOperands[i] = in canonicalizeMinMaxOp()
196 mlir::canonicalizeMapAndOperands(&newMap, &newOperands); in canonicalizeMinMaxOp()
198 rewriter.replaceOpWithNewOp<AffineApplyOp>(op, newMap, newOperands); in canonicalizeMinMaxOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp325 SmallVector<Value> newOperands; in matchAndRewrite() local
386 newOperands.push_back(validExtract in matchAndRewrite()
393 contractOp.getLoc(), newOperands[0], newOperands[1], newOperands[2], in matchAndRewrite()
418 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
421 newOperands.push_back(rewriter.create<vector::ExtractOp>( in matchAndRewrite()
424 newOperands.push_back(operand); in matchAndRewrite()
429 newOperands, newVecType, op->getAttrs()); in matchAndRewrite()
H A DVectorDistribute.cpp512 SmallVector<Value> newOperands(elementWise->getOperands().begin(), in matchAndRewrite() local
515 newOperands[i] = newWarpOp.getResult(newRetIndices[i]); in matchAndRewrite()
520 rewriter, loc, elementWise, newOperands, in matchAndRewrite()
792 SmallVector<Value> newOperands; in matchAndRewrite() local
799 newOperands.push_back(warpOp.getResult(yieldOperand.getOperandNumber())); in matchAndRewrite()
809 forOp.getStep(), newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DInlineScalarOperands.cpp37 SmallVector<Value> newOperands; in matchAndRewrite() local
44 newOperands.emplace_back(opOperand->get()); in matchAndRewrite()
57 loc, genericOp->getResultTypes(), newOperands, outputOperands, in matchAndRewrite()
H A DBufferizableOpInterfaceImpl.cpp65 SmallVector<Value> newOperands = newInputBuffers; in bufferizeLinalgOp() local
66 newOperands.append(newOutputBuffers.begin(), newOutputBuffers.end()); in bufferizeLinalgOp()
75 rewriter, op.getLoc(), /*resultTypes=*/TypeRange{}, newOperands)); in bufferizeLinalgOp()
H A DTransforms.cpp268 SmallVector<Value> newOperands; in rewriteAsPaddedOp() local
269 newOperands.reserve(opToPad.getNumInputsAndOutputs()); in rewriteAsPaddedOp()
276 newOperands.push_back(*paddedOperand); in rewriteAsPaddedOp()
288 ValueRange(newOperands).take_back(opToPad.getNumOutputs()).getTypes(); in rewriteAsPaddedOp()
289 paddedOp = opToPad.clone(b, loc, resultTensorTypes, newOperands); in rewriteAsPaddedOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferResultsToOutParams.cpp151 auto newOperands = llvm::to_vector<6>(op.getOperands()); in updateCalls() local
152 newOperands.append(outParams.begin(), outParams.end()); in updateCalls()
156 newResultTypes, newOperands); in updateCalls()
H A DFuncBufferizableOpInterfaceImpl.cpp277 SmallVector<Value> newOperands(numOperands, Value()); in bufferize() local
302 newOperands[idx] = tensorOperand; in bufferize()
307 Value buffer = newOperands[idx]; in bufferize()
331 newOperands[idx] = buffer; in bufferize()
336 callOp.getLoc(), funcOp.getSymName(), resultTypes, newOperands); in bufferize()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1063 OpOperand *newOperand = newOperands[it->second]; in populateOpPayload()
1668 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
1669 newOperands.reserve(op->getNumOperands()); in matchAndRewrite()
1673 newOperands.push_back(canFoldIntoConsumerOp(tensorCastOp) in matchAndRewrite()
1681 newOperands.push_back(fold ? tensorCastOp.getOperand() in matchAndRewrite()
1683 newResultTypes.push_back(newOperands.back().getType()); in matchAndRewrite()
1741 SmallVector<Value> newOperands = linalgOp.getInputOperands(); in matchAndRewrite() local
1810 newOperands.push_back(src); in createNewOperandWithStaticSizes()
1846 newOperands[index] = newOperand; in createNewOperandWithStaticSizes()
1878 SmallVector<Value> newOperands; in matchAndRewrite() local
[all …]
/llvm-project-15.0.7/mlir/lib/IR/
H A DOperationSupport.cpp194 void OperationState::addOperands(ValueRange newOperands) { in addOperands() argument
195 operands.append(newOperands.begin(), newOperands.end()); in addOperands()
344 MutableArrayRef<OpOperand> newOperands(newOperandStorage, newSize); in resize() local
347 newOperands.begin()); in resize()
355 new (&newOperands[numOperands]) OpOperand(owner); in resize()
365 return newOperands; in resize()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToSPIRV/
H A DVectorToSPIRV.cpp365 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
366 newOperands.reserve(oldResultType.getNumElements()); in matchAndRewrite()
368 newOperands.push_back(oldOperands[i.getZExtValue()]); in matchAndRewrite()
371 shuffleOp, newResultType, newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp2006 newOperands.append(ubOperands.begin(), ubOperands.end()); in setLowerBound()
2009 (*this)->setOperands(newOperands); in setLowerBound()
2022 (*this)->setOperands(newOperands); in setUpperBound()
2969 auto newOperands = in matchAndRewrite() local
3393 SmallVector<Value, 4> newOperands(lbOperands); in setLowerBounds() local
3395 (*this)->setOperands(newOperands); in setLowerBounds()
3406 (*this)->setOperands(newOperands); in setUpperBounds()
3482 SmallVector<Value, 4> newOperands{operands}; in canonicalize() local
3484 composeAffineMapAndOperands(&newMap, &newOperands); in canonicalize()
3485 if (newMap == getAffineMap() && newOperands == operands) in canonicalize()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp1419 SmallVector<Value, 4> newOperands(opInst->getOperands()); in createAffineComputationSlice() local
1420 for (unsigned i = 0, e = newOperands.size(); i < e; i++) { in createAffineComputationSlice()
1424 if (newOperands[i] == subOperands[j]) in createAffineComputationSlice()
1428 newOperands[i] = (*sliceOps)[j]; in createAffineComputationSlice()
1431 for (unsigned idx = 0, e = newOperands.size(); idx < e; idx++) { in createAffineComputationSlice()
1432 opInst->setOperand(idx, newOperands[idx]); in createAffineComputationSlice()
H A DLoopFusionUtils.cpp379 auto newOperands = forOp.getBody()->getTerminator()->getOperands(); in promoteSingleIterReductionLoop() local
384 b, parentForOp, iterOperands, newOperands, forOp.getRegionIterArgs()); in promoteSingleIterReductionLoop()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp756 SmallVector<Value> newOperands; in convertForOp() local
763 operand.index(), op.getNumIterOperands() + newOperands.size())); in convertForOp()
764 newOperands.push_back(it->second); in convertForOp()
767 scf::ForOp newForOp = replaceForOpWithNewSignature(b, op, newOperands); in convertForOp()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp520 SmallVector<Value, 4> newOperands(adaptor.getOperands()); in matchAndRewrite() local
529 newOperands[idx] = event; in matchAndRewrite()
536 [&] { yieldOp->setOperands(newOperands); }); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp692 auto newOperands = llvm::to_vector<8>( in matchAndRewrite() local
696 if (newOperands.size() < op.getNumOperands()) { in matchAndRewrite()
697 rewriter.replaceOpWithNewOp<OpTy>(op, op->getResultTypes(), newOperands, in matchAndRewrite()
794 auto newOperands = llvm::to_vector<8>( in matchAndRewrite() local
800 rewriter.replaceOpWithNewOp<OpTy>(op, op->getResultTypes(), newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp637 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
649 newOperands.push_back(*operandsIt++); in matchAndRewrite()
656 if (newOperands.size() == tensorFromElements.getDynamicExtents().size()) in matchAndRewrite()
662 newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DOperationSupport.h653 void addOperands(ValueRange newOperands);
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2775 SmallVector<Value, 4> newOperands; in matchAndRewrite() local
2800 newOperands.push_back(viewOp.getSizes()[dynamicDimPos]); in matchAndRewrite()
2815 viewOp.getByteShift(), newOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/docs/
H A DOpDefinitions.md1085 void process(AddOp op, ArrayRef<Value> newOperands) {
1087 zip(Adaptor<AddOp>(newOperands));