Lines Matching refs:symbolTableOp

53   Operation *symbolTableOp = symbol->getParentOp();  in collectValidReferencesFor()  local
54 if (within == symbolTableOp) in collectValidReferencesFor()
63 if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in collectValidReferencesFor()
67 getNameIfSymbol(symbolTableOp, symbolNameId); in collectValidReferencesFor()
72 symbolTableOp = symbolTableOp->getParentOp(); in collectValidReferencesFor()
73 if (symbolTableOp == within) in collectValidReferencesFor()
110 SymbolTable::SymbolTable(Operation *symbolTableOp) in SymbolTable() argument
111 : symbolTableOp(symbolTableOp) { in SymbolTable()
112 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>() && in SymbolTable()
114 assert(symbolTableOp->getNumRegions() == 1 && in SymbolTable()
116 assert(llvm::hasSingleElement(symbolTableOp->getRegion(0)) && in SymbolTable()
120 symbolTableOp->getContext()); in SymbolTable()
121 for (auto &op : symbolTableOp->getRegion(0).front()) { in SymbolTable()
143 assert(symbol->getParentOp() == symbolTableOp && in erase()
163 auto &body = symbolTableOp->getRegion(0).front(); in insert()
167 assert(insertPt->getParentOp() == symbolTableOp && in insert()
172 assert(symbol->getParentOp() == symbolTableOp && in insert()
291 Operation *SymbolTable::lookupSymbolIn(Operation *symbolTableOp, in lookupSymbolIn() argument
293 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>()); in lookupSymbolIn()
297 symbolTableOp->getContext()); in lookupSymbolIn()
298 for (auto &op : symbolTableOp->getRegion(0).front().without_terminator()) in lookupSymbolIn()
303 Operation *SymbolTable::lookupSymbolIn(Operation *symbolTableOp, in lookupSymbolIn() argument
306 if (failed(lookupSymbolIn(symbolTableOp, symbol, resolvedSymbols))) in lookupSymbolIn()
314 Operation *symbolTableOp, SymbolRefAttr symbol, in lookupSymbolInImpl() argument
317 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>()); in lookupSymbolInImpl()
320 symbolTableOp = lookupSymbolFn(symbolTableOp, symbol.getRootReference()); in lookupSymbolInImpl()
321 if (!symbolTableOp) in lookupSymbolInImpl()
323 symbols.push_back(symbolTableOp); in lookupSymbolInImpl()
331 if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in lookupSymbolInImpl()
337 symbolTableOp = lookupSymbolFn(symbolTableOp, ref.getValue()); in lookupSymbolInImpl()
338 if (!symbolTableOp || !symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in lookupSymbolInImpl()
340 symbols.push_back(symbolTableOp); in lookupSymbolInImpl()
342 symbols.push_back(lookupSymbolFn(symbolTableOp, symbol.getLeafReference())); in lookupSymbolInImpl()
347 SymbolTable::lookupSymbolIn(Operation *symbolTableOp, SymbolRefAttr symbol, in lookupSymbolIn() argument
349 auto lookupFn = [](Operation *symbolTableOp, StringRef symbol) { in lookupSymbolIn() argument
350 return lookupSymbolIn(symbolTableOp, symbol); in lookupSymbolIn()
352 return lookupSymbolInImpl(symbolTableOp, symbol, symbols, lookupFn); in lookupSymbolIn()
360 Operation *symbolTableOp = getNearestSymbolTable(from); in lookupNearestSymbolFrom() local
361 return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr; in lookupNearestSymbolFrom()
365 Operation *symbolTableOp = getNearestSymbolTable(from); in lookupNearestSymbolFrom() local
366 return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr; in lookupNearestSymbolFrom()
955 Operation *SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp, in lookupSymbolIn() argument
957 return getSymbolTable(symbolTableOp).lookup(symbol); in lookupSymbolIn()
959 Operation *SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp, in lookupSymbolIn() argument
962 if (failed(lookupSymbolIn(symbolTableOp, name, symbols))) in lookupSymbolIn()
970 SymbolTableCollection::lookupSymbolIn(Operation *symbolTableOp, in lookupSymbolIn() argument
973 auto lookupFn = [this](Operation *symbolTableOp, StringRef symbol) { in lookupSymbolIn() argument
974 return lookupSymbolIn(symbolTableOp, symbol); in lookupSymbolIn()
976 return lookupSymbolInImpl(symbolTableOp, name, symbols, lookupFn); in lookupSymbolIn()
985 Operation *symbolTableOp = SymbolTable::getNearestSymbolTable(from); in lookupNearestSymbolFrom() local
986 return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr; in lookupNearestSymbolFrom()
991 Operation *symbolTableOp = SymbolTable::getNearestSymbolTable(from); in lookupNearestSymbolFrom() local
992 return symbolTableOp ? lookupSymbolIn(symbolTableOp, symbol) : nullptr; in lookupNearestSymbolFrom()