Lines Matching refs:BlockArgument
95 DenseSet<BlockArgument> blockArgsToDetensor) in FunctionNonEntryBlockConversion()
110 for (BlockArgument blockArgument : block.getArguments()) { in matchAndRewrite()
132 const DenseSet<BlockArgument> blockArgsToDetensor;
203 DenseSet<BlockArgument> &blockArgsToDetensor) = 0;
213 const DenseSet<BlockArgument> &blockArgsToDetensor) { in computeBranchOpDetensoring()
255 DenseSet<BlockArgument> &blockArgsToDetensor) override { in compute()
317 if (currentItem.dyn_cast<BlockArgument>()) { in compute()
318 BlockArgument currentItemBlockArgument = in compute()
319 currentItem.cast<BlockArgument>(); in compute()
408 DenseSet<BlockArgument> blockArgsToRemove; in compute()
451 DenseSet<BlockArgument> &blockArgsToDetensor) override { in compute()
458 for (BlockArgument blockArgument : block.getArguments()) in compute()
470 DenseSet<BlockArgument> blockArgsToDetensor; in runOnOperation()
499 blockArgsToDetensor, [&](BlockArgument blockArgument) { in runOnOperation()