Searched refs:consumerIdx (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Fusion.cpp | 262 LinalgOp consumer, unsigned consumerIdx) { in fuse() argument
268 Value shapedOperand = consumer.getShapedOperand(consumerIdx); in fuse()
381 findFusableProducer(LinalgOp consumer, unsigned consumerIdx, in findFusableProducer() argument
390 [consumerIdx]( in findFusableProducer()
392 return elem.indexingOpView.operandIndex == consumerIdx; in findFusableProducer()
399 if (!isSameSubView(consumer.getBuffer(consumerIdx), consumedView)) in findFusableProducer()
428 unsigned consumerIdx, in fuseProducerOfBuffer() argument
431 findFusableProducer(consumer, consumerIdx, graph); in fuseProducerOfBuffer()
442 Value consumerView = consumer.getShapedOperand(consumerIdx); in fuseProducerOfBuffer()
459 auto fusedProducer = fuse(b, producerOp, producerIdx, consumer, consumerIdx); in fuseProducerOfBuffer()
[all …]
|
| D | FusionOnTensors.cpp | 31 unsigned consumerIdx) { in areTensorOpsFusable() argument
43 AffineMap consumerIndexMap = consumer.getIndexingMap(consumerIdx); in areTensorOpsFusable()
94 unsigned consumerIdx, unsigned nloops) { in generateFusedTensorOpRegion() argument
129 if (consumerArg.index() == consumerIdx + numConsumerIndices) { in generateFusedTensorOpRegion()
168 mapper.map(consumerBlock.getArgument(consumerIdx + numConsumerIndices), in generateFusedTensorOpRegion()
179 fuseTensorOpsImpl(LinalgOp producer, LinalgOp consumer, unsigned consumerIdx, in fuseTensorOpsImpl() argument
181 if (!areTensorOpsFusable(producer, consumer, consumerIdx)) in fuseTensorOpsImpl()
193 std::next(consumerOperands.begin(), consumerIdx)); in fuseTensorOpsImpl()
195 fusedOperands.append(std::next(consumerOperands.begin(), consumerIdx + 1), in fuseTensorOpsImpl()
206 std::next(consumerIndexMaps.begin(), consumerIdx)); in fuseTensorOpsImpl()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/ |
| D | Utils.h | 100 unsigned consumerIdx,
108 unsigned consumerIdx);
114 unsigned consumerIdx);
|