Searched refs:consumerOpOperand (Results 1 – 4 of 4) sorted by relevance
194 OpOperand &consumerOpOperand) { in fuse() argument197 Value shapedOperand = consumerOpOperand.get(); in fuse()277 findFusableProducer(OpOperand &consumerOpOperand, in findFusableProducer() argument337 findFusableProducer(consumerOpOperand, graph); in fuseProducerOfBuffer()346 if (consumerOpOperand.get().getParentBlock() == in fuseProducerOfBuffer()365 b.setInsertionPoint(consumerOpOperand.getOwner()); in fuseProducerOfBuffer()405 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor()426 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor()437 if (consumerOpOperand.get().getParentBlock() == in fuseProducerOfTensor()449 consumerOpOperand); in fuseProducerOfTensor()[all …]
81 if (!consumer.isInputTensor(consumerOpOperand)) in areElementwiseOpsFusable()117 if (operand == consumerOpOperand->get()) in areElementwiseOpsFusable()141 OpOperand *consumerOpOperand, in generateFusedElementwiseOpRegion() argument144 auto consumer = cast<GenericOp>(consumerOpOperand->getOwner()); in generateFusedElementwiseOpRegion()175 assert(consumer.isInputTensor(consumerOpOperand) && in generateFusedElementwiseOpRegion()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()[all …]
328 OpOperand *consumerOpOperand) { in fuseProducer() argument331 if (tiledRootAndFusedOpsLoops.count(consumerOpOperand->getOwner()) == 0) in fuseProducer()343 consumerOpOperand->get().getDefiningOp<tensor::ExtractSliceOp>(); in fuseProducer()348 LinalgOp consumerOp = consumerOpOperand->getOwner(); in fuseProducer()355 BlockArgument bbArg = consumerOp.getTiedBlockArgument(consumerOpOperand); in fuseProducer()374 consumerOpOperand, tiledRootAndFusedOpsLoops[consumerOp]); in fuseProducer()389 Type consumerOperandType = consumerOpOperand->get().getType(); in fuseProducer()
273 OpOperand &consumerOpOperand,281 OpOperand &consumerOpOperand);290 OpOperand &consumerOpOperand);394 FailureOr<LinalgOp> fuseProducer(OpBuilder &b, OpOperand *consumerOpOperand);