Searched refs:symbol_table_ (Results 1 – 5 of 5) sorted by relevance
82 symbol_table_(table), in RewriteTFRCallOp()179 const SymbolTable& symbol_table_; member in mlir::TFR::__anon1b0998e30111::RewriteTFRCallOp430 TFRFuncOp func = symbol_table_.lookup<TFRFuncOp>(external_callee_name); in matchAndRewrite()
810 SymbolTableCollection symbol_table_; member in mlir::TF::ShapeInference827 symbol_users_(symbol_table_, module), in ShapeInference()831 symbol_table_.getSymbolTable(module); in ShapeInference()886 dyn_cast_or_null<func::FuncOp>(call_op.resolveCallable(&symbol_table_)); in InferShapeForCall()935 op.ResolveThenFunction(&symbol_table_).getFunctionType().getResults(); in InferShapeForIf()937 op.ResolveElseFunction(&symbol_table_).getFunctionType().getResults(); in InferShapeForIf()2379 {if_op.ResolveThenFunction(&symbol_table_), in PropagateShapeIntoAttachedFunctions()2380 if_op.ResolveElseFunction(&symbol_table_)}, in PropagateShapeIntoAttachedFunctions()2395 while_op.ResolveBodyFunction(&symbol_table_) in PropagateShapeIntoAttachedFunctions()2400 {while_op.ResolveCondFunction(&symbol_table_), in PropagateShapeIntoAttachedFunctions()[all …]
604 const tables::SymbolTable& symbol_table() const { return symbol_table_; } in symbol_table()606 tables::SymbolTable* mutable_symbol_table() { return &symbol_table_; } in mutable_symbol_table()932 tables::SymbolTable symbol_table_{&string_pool_};
1192 symbol_table_(*symbol_table), in TFRTWhileOpConversion()1293 mlir::SymbolTable &symbol_table_; member in tensorflow::__anon783e79230111::TFRTWhileOpConversion1315 if (auto pred_fn = symbol_table_.lookup<mlir::func::FuncOp>(pred_fn_name)) { in GetPredicateFunction()1356 symbol_table_.insert(pred_fn); in GetPredicateFunction()1380 if (auto body_fn = symbol_table_.lookup<mlir::func::FuncOp>(body_fn_name)) { in GetWhileBodyFunction()1447 symbol_table_.insert(body_fn); in GetWhileBodyFunction()
3723 symbol_table_(module_.get()), in SavedModelSignatureDefImporterLite()3766 mlir::SymbolTable symbol_table_ ABSL_GUARDED_BY(symbol_table_mu_);3836 symbol_table_.insert(func.clone()); in MoveConvertedFunctionsToModule()