Lines Matching refs:tree
710 void Pattern::collectBoundSymbols(DagNode tree, SymbolInfoMap &infoMap, in collectBoundSymbols() argument
712 auto treeName = tree.getSymbol(); in collectBoundSymbols()
713 auto numTreeArgs = tree.getNumArgs(); in collectBoundSymbols()
715 if (tree.isNativeCodeCall()) { in collectBoundSymbols()
721 infoMap.bindValues(treeName, tree.getNumReturnsOfNativeCode()), in collectBoundSymbols()
732 if (auto treeArg = tree.getArgAsNestedDag(i)) { in collectBoundSymbols()
743 auto treeArgName = tree.getArgName(i); in collectBoundSymbols()
747 DagLeaf leaf = tree.getArgAsLeaf(i); in collectBoundSymbols()
774 if (tree.isOperation()) { in collectBoundSymbols()
775 auto &op = getDialectOp(tree); in collectBoundSymbols()
783 if (auto dagArg = tree.getArgAsNestedDag(i)) { in collectBoundSymbols()
809 auto collectSymbolInEither = [&](DagNode parent, DagNode tree, in collectBoundSymbols()
811 for (int i = 0; i < tree.getNumArgs(); ++i, ++opArgIdx) { in collectBoundSymbols()
812 if (DagNode subTree = tree.getArgAsNestedDag(i)) { in collectBoundSymbols()
815 auto argName = tree.getArgName(i); in collectBoundSymbols()
824 if (auto treeArg = tree.getArgAsNestedDag(i)) { in collectBoundSymbols()
826 collectSymbolInEither(tree, treeArg, opArgIdx); in collectBoundSymbols()
837 auto treeArgName = tree.getArgName(i); in collectBoundSymbols()
842 verifyBind(infoMap.bindOpArgument(tree, treeArgName, op, opArgIdx), in collectBoundSymbols()