Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp159 Block &prodBlock = prod.region().front(); in matchAndRewrite() local
164 unsigned num = prodBlock.getNumArguments(); in matchAndRewrite()
166 addArg(mapper, fusedBlock, prodBlock.getArgument(i)); in matchAndRewrite()
168 addArg(mapper, fusedBlock, prodBlock.getArgument(num - 1)); in matchAndRewrite()
170 auto *acc = prodBlock.getTerminator()->getOperand(0).getDefiningOp(); in matchAndRewrite()
174 for (auto &op : prodBlock.without_terminator()) in matchAndRewrite()