Lines Matching refs:consumerOpOperand

194                      OpOperand &consumerOpOperand) {  in fuse()  argument
197 Value shapedOperand = consumerOpOperand.get(); in fuse()
201 b, consumerOpOperand.getOwner()->getLoc(), shapedOperand, en.index()); in fuse()
277 findFusableProducer(OpOperand &consumerOpOperand, in findFusableProducer() argument
280 << consumerOpOperand.get() << " @" in findFusableProducer()
281 << consumerOpOperand.getOperandNumber() << " in " in findFusableProducer()
282 << *consumerOpOperand.getOwner() << "\n"); in findFusableProducer()
283 LinalgOp consumerOp = dyn_cast<LinalgOp>(consumerOpOperand.getOwner()); in findFusableProducer()
304 v == consumerOpOperand.get() && operandNum && in findFusableProducer()
305 *operandNum == consumerOpOperand.getOperandNumber(); in findFusableProducer()
320 isFusableInto(dependenceGraph, consumerOp, consumerOpOperand.get(), in findFusableProducer()
334 mlir::linalg::fuseProducerOfBuffer(OpBuilder &b, OpOperand &consumerOpOperand, in fuseProducerOfBuffer() argument
337 findFusableProducer(consumerOpOperand, graph); in fuseProducerOfBuffer()
346 if (consumerOpOperand.get().getParentBlock() == in fuseProducerOfBuffer()
357 auto subView = consumerOpOperand.get().getDefiningOp<memref::SubViewOp>(); in fuseProducerOfBuffer()
365 b.setInsertionPoint(consumerOpOperand.getOwner()); in fuseProducerOfBuffer()
367 << *consumerOpOperand.getOwner() << "\n"); in fuseProducerOfBuffer()
369 auto fusedProducer = fuse(b, producerOp, *producerMap, consumerOpOperand); in fuseProducerOfBuffer()
404 mlir::linalg::fuseProducerOfTensor(OpBuilder &b, OpOperand &consumerOpOperand) { in fuseProducerOfTensor() argument
405 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor()
412 return fuseProducerOfTensor(b, producerOpResult, consumerOpOperand); in fuseProducerOfTensor()
417 OpOperand &consumerOpOperand) { in fuseProducerOfTensor() argument
422 LinalgOp consumerOp = dyn_cast<LinalgOp>(consumerOpOperand.getOwner()); in fuseProducerOfTensor()
426 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor()
437 if (consumerOpOperand.get().getParentBlock() == in fuseProducerOfTensor()
449 consumerOpOperand); in fuseProducerOfTensor()
457 Type consumerType = consumerOpOperand.get().getType(); in fuseProducerOfTensor()
460 consumerOpOperand.set(def); in fuseProducerOfTensor()