| 52 | } |
| 53 | |
| 54 | Status Rewrite(std::unique_ptr<Graph>* graph) { |
| 55 | FunctionLibraryDefinition flib_def((*graph)->flib_def()); |
| 56 | GraphOptimizationPassOptions opt_options; |
| 57 | SessionOptions session_options = SessionOptionsWithInlining(); |
| 58 | opt_options.session_options = &session_options; |
| 59 | opt_options.graph = graph; |
| 60 | opt_options.flib_def = &flib_def; |
| 61 | LowerFunctionalOpsPass pass; |
| 62 | return pass.Run(opt_options); |
| 63 | } |
| 64 | |
| 65 | TEST(LowerIfOpTest, Simple) { |
| 66 | std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); |
no test coverage detected