Searched refs:consumerArg (Results 1 – 1 of 1) sorted by relevance
128 for (auto consumerArg : llvm::enumerate(consumerBlock.getArguments())) { in generateFusedTensorOpRegion() local129 if (consumerArg.index() == consumerIdx + numConsumerIndices) { in generateFusedTensorOpRegion()152 if (consumerArg.index() < numConsumerIndices) { in generateFusedTensorOpRegion()153 mapper.map(consumerArg.value(), in generateFusedTensorOpRegion()154 fusedBlock->getArgument(consumerArg.index())); in generateFusedTensorOpRegion()156 mapper.map(consumerArg.value(), in generateFusedTensorOpRegion()157 fusedBlock->addArgument(consumerArg.value().getType())); in generateFusedTensorOpRegion()