Home
last modified time | relevance | path

Searched refs:yieldValues (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp180 if (yieldValues.insert(std::get<0>(newRet))) { in moveRegionToNewWarpOpAndAppendReturns()
182 indices.push_back(yieldValues.size() - 1); in moveRegionToNewWarpOpAndAppendReturns()
215 Value yieldValues = yieldOperand.get(); in getWarpResult() local
216 Operation *definedOp = yieldValues.getDefiningOp(); in getWarpResult()
489 SmallVector<Value> yieldValues; in matchAndRewrite() local
506 yieldValues.push_back(operand.get()); in matchAndRewrite()
631 SmallVector<Value> yieldValues; in matchAndRewrite() local
640 if (yield.getNumOperands() == yieldValues.size()) in matchAndRewrite()
643 rewriter, warpOp, yieldValues, resultTypes); in matchAndRewrite()
662 SmallVector<Value> yieldValues; in matchAndRewrite() local
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DBufferizableOpInterfaceImpl.cpp231 SmallVector<OpOperand *> yieldValues = in bufferRelation() local
233 assert(yieldValues.size() == 2 && "expected 2 yield values"); in bufferRelation()
235 yieldValues[0]->get(), yieldValues[1]->get()); in bufferRelation()
453 SmallVector<Value> yieldValues; in resolveConflicts() local
458 yieldValues.push_back(value); in resolveConflicts()
466 yieldValues.push_back(*alloc); in resolveConflicts()
470 yieldOp, [&]() { yieldOp.getResultsMutable().assign(yieldValues); }); in resolveConflicts()
/llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp384 SmallVector<Value, 2> yieldValues = bodyBuilder(builder, result.location); in build() local
385 builder.create<AssumingYieldOp>(result.location, yieldValues); in build()
388 for (Value v : yieldValues) in build()