| 152 | } |
| 153 | |
| 154 | void NodeMap::UpdateOutput(const string& node_name, |
| 155 | const string& old_output_name, |
| 156 | const string& new_output_name) { |
| 157 | std::set<NodeDef*>& outputs = outputs_[node_name]; |
| 158 | outputs.erase(nodes_[NodeName(old_output_name)]); |
| 159 | outputs.insert(nodes_[NodeName(new_output_name)]); |
| 160 | } |
| 161 | |
| 162 | string TensorIdToString(const TensorId& tensor_id) { |
| 163 | return tensor_id.index() == 0 ? string(tensor_id.node()) |
no test coverage detected