| 31 | namespace { |
| 32 | |
| 33 | Status ClusterScoping(std::unique_ptr<Graph>* graph) { |
| 34 | FixupSourceAndSinkEdges(graph->get()); |
| 35 | |
| 36 | GraphOptimizationPassOptions opt_options; |
| 37 | opt_options.graph = graph; |
| 38 | FunctionDefLibrary fdef_lib; |
| 39 | FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); |
| 40 | opt_options.flib_def = &flib_def; |
| 41 | SessionOptions session_options; |
| 42 | session_options.env = Env::Default(); |
| 43 | session_options.config.mutable_graph_options() |
| 44 | ->mutable_optimizer_options() |
| 45 | ->set_global_jit_level(OptimizerOptions::ON_2); |
| 46 | opt_options.session_options = &session_options; |
| 47 | |
| 48 | ClusterScopingPass pass; |
| 49 | return pass.Run(opt_options); |
| 50 | } |
| 51 | |
| 52 | absl::flat_hash_map<string, string> GetXlaInternalScopes(const Graph& graph) { |
| 53 | absl::flat_hash_map<string, string> scopes; |
no test coverage detected