Home
last modified time | relevance | path

Searched refs:initArgNumber (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DHoisting.cpp284 unsigned initArgNumber = tensorBBArg.getArgNumber() - /*numIvs=*/1; in hoistReadWrite() local
289 forOp.getInitArgs()[initArgNumber]); in hoistReadWrite()
292 forOp.getInitArgs()[initArgNumber]); in hoistReadWrite()
302 yieldOp->setOperand(initArgNumber, write.insertSliceOp.getDest()); in hoistReadWrite()
304 yieldOp->setOperand(initArgNumber, write.transferWriteOp.getSource()); in hoistReadWrite()
321 newForOp.getResult(initArgNumber) in hoistReadWrite()
328 newForOp.getResult(initArgNumber) in hoistReadWrite()
331 newForOp.getResult(initArgNumber)); in hoistReadWrite()