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