Searched refs:nodes_map (Results 1 – 2 of 2) sorted by relevance
312 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode() argument313 nodes_map_ = nodes_map; in TFGraphNode()318 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode() argument319 nodes_map_ = nodes_map; in TFGraphNode()383 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes_map) { in ToProto() argument404 (*node_.mutable_inputs())[inp.first] = nodes_map.at(inp.second)->id(); in ToProto()425 int64 id = nodes_map.at(s.first)->id(); in ToProto()
853 std::unordered_map<size_t, size_t> nodes_map; in PruneTree() local854 nodes_map[0] = 0; in PruneTree()863 auto mapped_node_it = nodes_map.find(node_idx); in PruneTree()864 CHECK(mapped_node_it != nodes_map.end()); in PruneTree()873 nodes_map[child_idx] = new_idx; in PruneTree()