Lines Matching refs:consumerOpOperand
69 OpOperand *consumerOpOperand) { in areElementwiseOpsFusable() argument
81 if (!consumer.isInputTensor(consumerOpOperand)) in areElementwiseOpsFusable()
86 AffineMap consumerIndexMap = consumer.getTiedIndexingMap(consumerOpOperand); in areElementwiseOpsFusable()
117 if (operand == consumerOpOperand->get()) in areElementwiseOpsFusable()
141 OpOperand *consumerOpOperand, in generateFusedElementwiseOpRegion() argument
143 auto producer = cast<GenericOp>(consumerOpOperand->get().getDefiningOp()); in generateFusedElementwiseOpRegion()
144 auto consumer = cast<GenericOp>(consumerOpOperand->getOwner()); in generateFusedElementwiseOpRegion()
175 assert(consumer.isInputTensor(consumerOpOperand) && in generateFusedElementwiseOpRegion()
179 consumerOpOperand->getOperandNumber())) // input assumption. in generateFusedElementwiseOpRegion()
204 .drop_front(consumerOpOperand->getOperandNumber() + 1)) in generateFusedElementwiseOpRegion()
238 mapper.map(consumerBlock.getArgument(consumerOpOperand->getOperandNumber()), in generateFusedElementwiseOpRegion()
250 fuseElementwiseOpsImpl(GenericOp producer, OpOperand *consumerOpOperand, in fuseElementwiseOpsImpl() argument
253 auto consumer = cast<GenericOp>(consumerOpOperand->getOwner()); in fuseElementwiseOpsImpl()
254 if (!areElementwiseOpsFusable(producer, consumer, consumerOpOperand) || in fuseElementwiseOpsImpl()
255 !controlFn(producer->getResult(0), *consumerOpOperand)) in fuseElementwiseOpsImpl()
259 assert(consumer.isInputTensor(consumerOpOperand) && in fuseElementwiseOpsImpl()
273 llvm::find(consumerInputs, consumerOpOperand); in fuseElementwiseOpsImpl()
288 consumer.getTiedIndexingMap(consumerOpOperand)); in fuseElementwiseOpsImpl()
299 consumer.getTiedIndexingMap(consumerOpOperand)); in fuseElementwiseOpsImpl()
337 consumer.getTiedIndexingMap(consumerOpOperand); in fuseElementwiseOpsImpl()
349 consumerOpOperand, consumer.getNumLoops()); in fuseElementwiseOpsImpl()
354 fuseElementwiseOps(PatternRewriter &rewriter, OpOperand *consumerOpOperand, in fuseElementwiseOps() argument
359 return fuseElementwiseOpsImpl(producer, consumerOpOperand, controlFn, in fuseElementwiseOps()