Lines Matching refs:opOperand
240 replaceUnitExtents(GenericOp genericOp, OpOperand *opOperand, in replaceUnitExtents() argument
242 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in replaceUnitExtents()
243 ArrayRef<int64_t> shape = genericOp.getShape(opOperand); in replaceUnitExtents()
250 int64_t origRank = genericOp.getRank(opOperand); in replaceUnitExtents()
258 Type actualType = opOperand->get().getType(); in replaceUnitExtents()
284 Type elementType = getElementTypeOrSelf(opOperand->get()); in replaceUnitExtents()
286 if (elementType == opOperand->get().getType()) { in replaceUnitExtents()
373 for (OpOperand *opOperand : genericOp.getInputAndOutputOperands()) { in matchAndRewrite()
374 auto replacementInfo = replaceUnitExtents(genericOp, opOperand, context); in matchAndRewrite()
380 replacementInfo->type != opOperand->get().getType(); in matchAndRewrite()
385 newInputOutputTypes.push_back(opOperand->get().getType()); in matchAndRewrite()
386 newIndexingMaps.push_back(genericOp.getTiedIndexingMap(opOperand)); in matchAndRewrite()
387 int64_t origRank = genericOp.getRank(opOperand); in matchAndRewrite()