Lines Matching refs:opOperand

275   for (OpOperand *opOperand : llvm::make_range(consumerInputs.begin(), it)) {  in fuseElementwiseOpsImpl()
276 fusedOperands.push_back(opOperand->get()); in fuseElementwiseOpsImpl()
277 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
283 for (OpOperand *opOperand : producer.getInputOperands()) { in fuseElementwiseOpsImpl()
284 fusedOperands.push_back(opOperand->get()); in fuseElementwiseOpsImpl()
287 opOperand, producerResultIndexMap, in fuseElementwiseOpsImpl()
304 for (OpOperand *opOperand : in fuseElementwiseOpsImpl()
306 fusedOperands.push_back(opOperand->get()); in fuseElementwiseOpsImpl()
307 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
310 for (OpOperand *opOperand : consumer.getOutputOperands()) in fuseElementwiseOpsImpl()
311 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
375 for (OpOperand *opOperand : genericOp.getInputAndOutputOperands()) { in matchAndRewrite()
377 dyn_cast_or_null<GenericOp>(opOperand->get().getDefiningOp()); in matchAndRewrite()
381 fuseElementwiseOps(rewriter, opOperand, producer, controlFn); in matchAndRewrite()
718 for (OpOperand *opOperand : genericOp.getInputOperands()) { in fuseWithReshapeByExpansion()
719 if (opOperand == fusableOpOperand) { in fuseWithReshapeByExpansion()
724 if (genericOp.isInputTensor(opOperand)) { in fuseWithReshapeByExpansion()
725 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion()
726 auto opOperandType = opOperand->get().getType().cast<RankedTensorType>(); in fuseWithReshapeByExpansion()
729 if (expandedOperandType != opOperand->get().getType()) { in fuseWithReshapeByExpansion()
742 genericOp.getLoc(), expandedOperandType, opOperand->get(), in fuseWithReshapeByExpansion()
747 expandedOpOperands.push_back(opOperand->get()); in fuseWithReshapeByExpansion()
752 for (OpOperand *opOperand : genericOp.getOutputOperands()) { in fuseWithReshapeByExpansion()
753 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion()
754 auto opOperandType = opOperand->get().getType().cast<RankedTensorType>(); in fuseWithReshapeByExpansion()
757 if (expandedOutputType != opOperand->get().getType()) { in fuseWithReshapeByExpansion()
769 genericOp.getLoc(), expandedOutputType, opOperand->get(), in fuseWithReshapeByExpansion()
828 for (OpOperand *opOperand : genericOp.getInputTensorOperands()) { in matchAndRewrite()
830 opOperand->get().getDefiningOp<tensor::CollapseShapeOp>(); in matchAndRewrite()
836 if (!isFusableWithReshapeByDimExpansion(genericOp, opOperand) || in matchAndRewrite()
837 (!controlFoldingReshapes(reshapeOp->getResult(0), *opOperand))) in matchAndRewrite()
841 fuseWithReshapeByExpansion(genericOp, reshapeOp, opOperand, rewriter); in matchAndRewrite()
1272 OpOperand *opOperand, in getCollapsedOpOperand() argument
1275 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in getCollapsedOpOperand()
1281 Value operand = opOperand->get(); in getCollapsedOpOperand()
1363 llvm::map_range(genericOp.getInputOperands(), [&](OpOperand *opOperand) { in collapseGenericOpIterationDims() argument
1364 return getCollapsedOpOperand(loc, genericOp, opOperand, collapsingInfo, in collapseGenericOpIterationDims()
1448 for (OpOperand *opOperand : genericOp.getInputTensorOperands()) { in matchAndRewrite()
1450 opOperand->get().getDefiningOp<tensor::ExpandShapeOp>(); in matchAndRewrite()
1455 getCollapsableIterationSpaceDims(genericOp, opOperand, in matchAndRewrite()
1458 !controlFoldingReshapes(reshapeOp->getResult(0), *opOperand)) { in matchAndRewrite()
1464 opOperand, rewriter); in matchAndRewrite()
1497 for (OpOperand *opOperand : genericOp.getInputOperands()) { in matchAndRewrite()
1498 Operation *def = opOperand->get().getDefiningOp(); in matchAndRewrite()
1527 auto resultValue = opOperand->get().dyn_cast<OpResult>(); in matchAndRewrite()
1541 if (inputOperand == opOperand) in matchAndRewrite()
1576 mapping.map(entryBlock.getArgument(opOperand->getOperandNumber()), in matchAndRewrite()
1607 for (OpOperand *opOperand : op.getOutputOperands()) { in matchAndRewrite()
1608 if (!op.payloadUsesValueFromOperand(opOperand)) { in matchAndRewrite()
1609 Value operandVal = opOperand->get(); in matchAndRewrite()
1633 op->setOperand(opOperand->getOperandNumber(), initTensor); in matchAndRewrite()
1655 for (OpOperand *opOperand : genericOp.getInputOperands()) { in matchAndRewrite()
1656 if (!genericOp.payloadUsesValueFromOperand(opOperand)) in matchAndRewrite()
1658 FillOp fillOp = opOperand->get().getDefiningOp<FillOp>(); in matchAndRewrite()
1662 payload.getArgument(opOperand->getOperandNumber()) in matchAndRewrite()