| 77 | } |
| 78 | |
| 79 | Node* GraphDefBuilder::Options::FinalizeBuilder(NodeBuilder* builder) const { |
| 80 | builder->ControlInputs(control_inputs_); |
| 81 | if (!device_.empty()) builder->Device(device_); |
| 82 | for (const auto& attr : attrs_) { |
| 83 | builder->Attr(attr.first, attr.second); |
| 84 | } |
| 85 | |
| 86 | Node* returned_node; |
| 87 | UpdateStatus(builder->Finalize(graph_, &returned_node)); |
| 88 | return returned_node; |
| 89 | } |
| 90 | |
| 91 | void GraphDefBuilder::Options::UpdateStatus(const Status& status) const { |
| 92 | if (status_ == nullptr) { |