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