MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsTrainingGraph

Method IsTrainingGraph

tensorflow/core/graph/graph.cc:845–852  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

843}
844
845bool 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
854std::unordered_map<string, Node*> Graph::BuildNodeNameIndex() const {
855 std::unordered_map<string, Node*> result;

Callers 2

MarkEmbeddingGraphFunction · 0.80

Calls 2

nameMethod · 0.65
findMethod · 0.45

Tested by

no test coverage detected