Home
last modified time | relevance | path

Searched refs:reachable_funcs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_identity_pruning.cc50 llvm::SmallPtrSet<FuncOp, 4> reachable_funcs; in CollectReachableFunctions() local
53 [&reachable_funcs](Region& src, SmallVectorImpl<FuncOp>& funcs_to_visit) { in CollectReachableFunctions()
54 src.walk([&reachable_funcs, &funcs_to_visit](CallOpInterface call_op) { in CollectReachableFunctions()
56 if (func && reachable_funcs.insert(func).second) in CollectReachableFunctions()
74 return llvm::to_vector<4>(reachable_funcs); in CollectReachableFunctions()
95 auto reachable_funcs = CollectReachableFunctions(cluster.body()); in runOnOperation() local
96 for (FuncOp reachable_func : reachable_funcs) in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dtrim_functions_tf.cc107 llvm::SetVector<FuncOp> reachable_funcs; in Verify() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1613 std::set<string> reachable_funcs; in ReachableFunctions() local
1628 if (func && reachable_funcs.find(func_name) == reachable_funcs.end()) { in ReachableFunctions()
1681 reachable_funcs.insert(func_name); in ReachableFunctions()
1697 return reachable_funcs; in ReachableFunctions()
1703 std::set<string> reachable_funcs = ReachableFunctions(flib, nodes); in ReachableFunctionLibraryDefinition() local
1708 for (const string& func_name : reachable_funcs) { in ReachableFunctionLibraryDefinition()