Searched refs:all_nodes_ (Results 1 – 2 of 2) sorted by relevance
114 all_nodes_.insert( in Initialize()124 all_nodes_.insert(std::make_pair(div_const_node->name(), div_const_node)); in Initialize()137 auto apply_gradients_op = all_nodes_[apply_gradient_node_name]->op(); in Initialize()138 auto apply_gradients_node = all_nodes_[apply_gradient_node_name]; in Initialize()144 all_nodes_.insert(std::make_pair(div_node->name(), div_node)); in Initialize()148 LOG(INFO) << "Graph size after adding div nodes: " << all_nodes_.size(); in Initialize()191 for (const auto& node : all_nodes_) { in Initialize()208 *new_node = *all_nodes_[node]; in AddSharedNodes()222 *new_node = *all_nodes_[node]; in AddOneReplica()
44 std::map<string, NodeDef*> all_nodes_; variable