- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for SymbolUserMap (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/guarantee_all_funcs_one_use.cc
bool made_changes = false; if (failed(CheckNoRecursion(module, getAnalysis<CallGraph>()))) return failure(); do { SymbolUserMap symbol_users(symbol_table_collection, module); made_changes = false; for (auto func : llvm::make_early_inc_range(module.getOps<func::FuncOp>())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
// device cluster ops. void PopulateClusterReachableFunctions( ModuleOp module, SmallPtrSetImpl<Operation*>& reachable_functions) { SymbolTableCollection table; SymbolUserMap symbol_map(table, module); // Create map from caller to set of all callee(s). llvm::DenseMap<func::FuncOp, llvm::DenseSet<func::FuncOp>> caller_callee_map;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
module.walk([&](Operation* op) { if (IsCommunicationOp(op)) { ops_with_tokens.insert(op); worklist.push_back(op); } }); SymbolTableCollection table; SymbolUserMap symbol_map(table, module); // Regions that contains ops requiring token input attributes. absl::flat_hash_set<Region*> regions_with_token; while (!worklist.empty()) { Operation* op = worklist.back();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
// the uses is as a while condition, an empty vector is returned. SmallVector<TF::WhileOp> GetWhileCallers(func::FuncOp func, SymbolUserMap& symbol_map) { SmallVector<TF::WhileOp> while_callers; for (auto user : symbol_map.getUsers(func)) { if (auto while_caller = dyn_cast<TF::WhileOp>(user)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
} // Returns all functions that can be reached from TPUPartitionedCall ops. SmallPtrSet<Operation*, 16> FindTPUPartitionedCallReachableFunctions( ModuleOp module) { SymbolTableCollection table; SymbolUserMap symbol_map(table, module); llvm::DenseMap<func::FuncOp, llvm::DenseSet<func::FuncOp>> caller_callee_map; // Creates work queue for determining reachability below. std::queue<func::FuncOp> function_worklist;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// corresponds to a constant value. ValuePortResultMap results_; // Map from a function to the callers of that function. SymbolTableCollection symbol_table_; SymbolUserMap symbol_users_; // Queue of functions being processed. llvm::DenseSet<func::FuncOp> queue_set_; std::queue<func::FuncOp> queue_; int64_t graph_version_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)