| 544 | } |
| 545 | |
| 546 | GraphOptimizer::GraphOptimizer( |
| 547 | const std::string& signature_name, |
| 548 | MetaGraphDef* mgdef, |
| 549 | GraphOptimizerOption& option) |
| 550 | : graph_(OpRegistry::Global()), |
| 551 | signature_name_(signature_name), |
| 552 | meta_graph_def_(mgdef), |
| 553 | option_(option) { |
| 554 | |
| 555 | GraphConstructorOptions opts; |
| 556 | opts.allow_internal_ops = true; |
| 557 | opts.allow_internal_ops = true; |
| 558 | |
| 559 | // Add default attributes |
| 560 | GraphDef converted_graph_def; |
| 561 | Status s = processor::AddDefaultAttributes( |
| 562 | meta_graph_def_->graph_def(), &converted_graph_def); |
| 563 | if (!s.ok()) { |
| 564 | LOG(WARNING) << "Add default attrs to graphdef failed. " |
| 565 | << s.error_message(); |
| 566 | } |
| 567 | s = ConvertGraphDefToGraph( |
| 568 | opts, converted_graph_def/*meta_graph_def_->graph_def()*/, &graph_); |
| 569 | if (!s.ok()) { |
| 570 | LOG(FATAL) << "Can not convert graphdef to graph, " |
| 571 | << s.error_message(); |
| 572 | } |
| 573 | } |
| 574 | |
| 575 | |
| 576 | SavedModelOptimizer::SavedModelOptimizer( |
nothing calls this directly
no test coverage detected