Lines Matching refs:blockArgsToDetensor
95 DenseSet<BlockArgument> blockArgsToDetensor) in FunctionNonEntryBlockConversion()
97 blockArgsToDetensor(std::move(blockArgsToDetensor)) {} in FunctionNonEntryBlockConversion()
113 if (blockArgsToDetensor.count(blockArgument)) in matchAndRewrite()
132 const DenseSet<BlockArgument> blockArgsToDetensor; member
203 DenseSet<BlockArgument> &blockArgsToDetensor) = 0;
213 const DenseSet<BlockArgument> &blockArgsToDetensor) { in computeBranchOpDetensoring() argument
216 for (auto blockArgumentElem : blockArgsToDetensor) { in computeBranchOpDetensoring()
255 DenseSet<BlockArgument> &blockArgsToDetensor) override { in compute() argument
285 if (succBlockArg && !blockArgsToDetensor.count(*succBlockArg)) in compute()
328 blockArgsToDetensor.insert(currentItemBlockArgument); in compute()
339 blockArgsToDetensor.clear(); in compute()
410 for (auto &blockArg : blockArgsToDetensor) { in compute()
440 blockArgsToDetensor.erase(blockArg); in compute()
451 DenseSet<BlockArgument> &blockArgsToDetensor) override { in compute() argument
459 blockArgsToDetensor.insert(blockArgument); in compute()
470 DenseSet<BlockArgument> blockArgsToDetensor; in runOnOperation() local
476 blockArgsToDetensor); in runOnOperation()
480 blockArgsToDetensor); in runOnOperation()
484 CostModel::computeBranchOpDetensoring(blockArgsToDetensor); in runOnOperation()
499 blockArgsToDetensor, [&](BlockArgument blockArgument) { in runOnOperation()
528 blockArgsToDetensor); in runOnOperation()