| 765 | } |
| 766 | |
| 767 | void Model::RemoveNode(const string& name) { |
| 768 | mutex_lock l(mu_); |
| 769 | auto node = gtl::FindOrNull(lookup_table_, name); |
| 770 | if (node) { |
| 771 | if ((*node)->output()) { |
| 772 | (*node)->output()->remove_input(*node); |
| 773 | } |
| 774 | VLOG(3) << "Removing " << (*node)->long_name(); |
| 775 | remove_node_hook_(*node); |
| 776 | } |
| 777 | lookup_table_.erase(name); |
| 778 | } |
| 779 | |
| 780 | std::map<string, std::shared_ptr<Parameter>> Model::CollectTunableParameters( |
| 781 | std::shared_ptr<Node> node) { |
no test coverage detected