Lines Matching defs:tree

152   // We want to get number of operations recursively involved in the DAG tree.
734 void Pattern::collectBoundSymbols(DagNode tree, SymbolInfoMap &infoMap,
736   auto treeName = tree.getSymbol();
737   auto numTreeArgs = tree.getNumArgs();
739   if (tree.isNativeCodeCall()) {
745             infoMap.bindValues(treeName, tree.getNumReturnsOfNativeCode()),
756       if (auto treeArg = tree.getArgAsNestedDag(i)) {
767       auto treeArgName = tree.getArgName(i);
771         DagLeaf leaf = tree.getArgAsLeaf(i);
798   if (tree.isOperation()) {
799     auto &op = getDialectOp(tree);
807       if (auto dagArg = tree.getArgAsNestedDag(i)) {
833     auto collectSymbolInEither = [&](DagNode parent, DagNode tree,
835       for (int i = 0; i < tree.getNumArgs(); ++i, ++opArgIdx) {
836         if (DagNode subTree = tree.getArgAsNestedDag(i)) {
839           auto argName = tree.getArgName(i);
851     auto collectSymbolInVariadic = [&](DagNode parent, DagNode tree,
853       auto treeName = tree.getSymbol();
861       for (int i = 0; i < tree.getNumArgs(); ++i) {
862         if (DagNode subTree = tree.getArgAsNestedDag(i)) {
865           auto argName = tree.getArgName(i);
876       if (auto treeArg = tree.getArgAsNestedDag(i)) {
878           collectSymbolInEither(tree, treeArg, opArgIdx);
888           collectSymbolInVariadic(tree, treeArg, opArgIdx);
899         auto treeArgName = tree.getArgName(i);
904           verifyBind(infoMap.bindOpArgument(tree, treeArgName, op, opArgIdx),