Searched refs:grad_nodes_ (Results 1 – 2 of 2) sorted by relevance
90 std::map<string, std::vector<TFGraphNode*>> grad_nodes_; variable
401 auto grad_nodes_it = grad_nodes_.find(forward_name); in AddNode()402 if (grad_nodes_it != grad_nodes_.end()) { in AddNode()405 grad_nodes_.insert( in AddNode()438 for (const auto& it : grad_nodes_) { in Build()