| 29 | namespace tensorflow { |
| 30 | |
| 31 | bool OptimizeFusion(Graph* g) { |
| 32 | |
| 33 | bool changed = false; |
| 34 | std::vector<std::unique_ptr<TemplateBase>> templates; |
| 35 | templates.emplace_back(new TemplateSparseInnerFlatten()); |
| 36 | templates.emplace_back(new TemplateLogicSumBase()); |
| 37 | templates.emplace_back(new TemplateSelectPruningElseConst()); |
| 38 | templates.emplace_back(new TemplateSelectPruningThenConst()); |
| 39 | templates.emplace_back(new TemplateSelectThenScalar()); |
| 40 | templates.emplace_back(new TemplateSelectElseScalar()); |
| 41 | templates.emplace_back(new TemplateSelectElseScalarInGrad()); |
| 42 | templates.emplace_back(new TemplateSelectThenScalarInGrad()); |
| 43 | |
| 44 | for (auto& t : templates) { |
| 45 | std::unique_ptr<OptimizerFusionImpl> opt( |
| 46 | new OptimizerFusionImpl(g, t.get())); |
| 47 | changed |= opt->Optimize(); |
| 48 | } |
| 49 | |
| 50 | return changed; |
| 51 | } |
| 52 | |
| 53 | } // namespace tensorflow |