| 965 | } |
| 966 | |
| 967 | Tensor* Graph::GetOutputTensor(const std::string& name) |
| 968 | { |
| 969 | for(unsigned int i = 0; i < output_nodes.size(); i++) |
| 970 | { |
| 971 | Node* node = output_nodes[i]; |
| 972 | |
| 973 | for(unsigned int j = 0; j < node->GetOutputNum(); j++) |
| 974 | { |
| 975 | Tensor* tensor = node->GetOutputTensor(j); |
| 976 | |
| 977 | if(tensor->GetName() == name) |
| 978 | return tensor; |
| 979 | } |
| 980 | } |
| 981 | |
| 982 | return nullptr; |
| 983 | } |
| 984 | |
| 985 | bool Graph::Replace(Subgraph* orig_sub, Subgraph* new_sub) |
| 986 | { |