| 843 | } |
| 844 | |
| 845 | bool Graph::IsTrainingGraph() const { |
| 846 | for (Node* node : op_nodes()) { |
| 847 | if (node->name().find("gradient") != std::string::npos) { |
| 848 | return true; |
| 849 | } |
| 850 | } |
| 851 | return false; |
| 852 | } |
| 853 | |
| 854 | std::unordered_map<string, Node*> Graph::BuildNodeNameIndex() const { |
| 855 | std::unordered_map<string, Node*> result; |
no test coverage detected