49   Operation *symbolTableOp = symbol->getParentOp();  in collectValidReferencesFor()  local
50   if (within == symbolTableOp)  in collectValidReferencesFor()
59     if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>())  in collectValidReferencesFor()
62     StringAttr symbolTableName = getNameIfSymbol(symbolTableOp, symbolNameId);  in collectValidReferencesFor()
67     symbolTableOp = symbolTableOp->getParentOp();  in collectValidReferencesFor()
68     if (symbolTableOp == within)  in collectValidReferencesFor()
117 SymbolTable::SymbolTable(Operation *symbolTableOp)  in SymbolTable()  argument
118     : symbolTableOp(symbolTableOp) {  in SymbolTable()
119   assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>() &&  in SymbolTable()
121   assert(symbolTableOp->getNumRegions() == 1 &&  in SymbolTable()
123   assert(llvm::hasSingleElement(symbolTableOp->getRegion(0)) &&  in SymbolTable()
126   StringAttr symbolNameId = StringAttr::get(symbolTableOp->getContext(),  in SymbolTable()
128   for (auto &op : symbolTableOp->getRegion(0).front()) {  in SymbolTable()
143   return lookup(StringAttr::get(symbolTableOp->getContext(), name));  in lookup()
153   assert(symbol->getParentOp() == symbolTableOp &&  in erase()
174     auto &body = symbolTableOp->getRegion(0).front();  in insert()
179               insertPt->getParentOp() == symbolTableOp) &&  in insert()
189   assert(symbol->getParentOp() == symbolTableOp &&  in insert()
312 Operation *SymbolTable::lookupSymbolIn(Operation *symbolTableOp,  in lookupSymbolIn()  argument
314   assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>());  in lookupSymbolIn()
315   Region &region = symbolTableOp->getRegion(0);  in lookupSymbolIn()
320   StringAttr symbolNameId = StringAttr::get(symbolTableOp->getContext(),  in lookupSymbolIn()
327 Operation *SymbolTable::lookupSymbolIn(Operation *symbolTableOp,  in lookupSymbolIn()  argument
330   if (failed(lookupSymbolIn(symbolTableOp, symbol, resolvedSymbols)))  in lookupSymbolIn()
338     Operation *symbolTableOp, SymbolRefAttr symbol,  in lookupSymbolInImpl()  argument
341   assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>());  in lookupSymbolInImpl()
344   symbolTableOp = lookupSymbolFn(symbolTableOp, symbol.getRootReference());  in lookupSymbolInImpl()
345   if (!symbolTableOp)  in lookupSymbolInImpl()
347   symbols.push_back(symbolTableOp);  in lookupSymbolInImpl()
355   if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>())  in lookupSymbolInImpl()
361     symbolTableOp = lookupSymbolFn(symbolTableOp, ref.getAttr());  in lookupSymbolInImpl()
362     if (!symbolTableOp || !symbolTableOp->hasTrait<OpTrait::SymbolTable>())  in lookupSymbolInImpl()
364     symbols.push_back(symbolTableOp);  in lookupSymbolInImpl()
366   symbols.push_back(lookupSymbolFn(symbolTableOp, symbol.getLeafReference()));  in lookupSymbolInImpl()
371 SymbolTable::lookupSymbolIn(Operation *symbolTableOp, SymbolRefAttr symbol,  in lookupSymbolIn()  argument
373   auto lookupFn = [](Operation *symbolTableOp, StringAttr symbol) {  in lookupSymbolIn()  argument
374     return lookupSymbolIn(symbolTableOp, symbol);  in lookupSymbolIn()
376   return lookupSymbolInImpl(symbolTableOp, symbol, symbols, lookupFn);  in lookupSymbolIn()
384   Operation *symbolTableOp = getNearestSymbolTable(from);  in lookupNearestSymbolFrom()  local
385   return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr;  in lookupNearestSymbolFrom()
389   Operation *symbolTableOp = getNearestSymbolTable(from);  in lookupNearestSymbolFrom()  local
390   return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr;  in lookupNearestSymbolFrom()
919 Operation *SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp,  in lookupSymbolIn()  argument
921   return getSymbolTable(symbolTableOp).lookup(symbol);  in lookupSymbolIn()
923 Operation *SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp,  in lookupSymbolIn()  argument
926   if (failed(lookupSymbolIn(symbolTableOp, name, symbols)))  in lookupSymbolIn()
934 SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp,  in lookupSymbolIn()  argument
937   auto lookupFn = [this](Operation *symbolTableOp, StringAttr symbol) {  in lookupSymbolIn()  argument
938     return lookupSymbolIn(symbolTableOp, symbol);  in lookupSymbolIn()
940   return lookupSymbolInImpl(symbolTableOp, name, symbols, lookupFn);  in lookupSymbolIn()
949   Operation *symbolTableOp = SymbolTable::getNearestSymbolTable(from);  in lookupNearestSymbolFrom()  local
950   return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr;  in lookupNearestSymbolFrom()
955   Operation *symbolTableOp = SymbolTable::getNearestSymbolTable(from);  in lookupNearestSymbolFrom()  local
956   return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr;  in lookupNearestSymbolFrom()
972                              Operation *symbolTableOp)  in SymbolUserMap()  argument
976   auto walkFn = [&](Operation *symbolTableOp, bool allUsesVisible) {  in SymbolUserMap()  argument
977     for (Operation &nestedOp : symbolTableOp->getRegion(0).getOps()) {  in SymbolUserMap()
983         (void)symbolTable.lookupSymbolIn(symbolTableOp, use.getSymbolRef(),  in SymbolUserMap()
992   SymbolTable::walkSymbolTables(symbolTableOp, /*allSymUsesVisible=*/false,  in SymbolUserMap()