Searched refs:replicated_nodes_map_ (Results 1 – 1 of 1) sorted by relevance
30 if (replicated_nodes_map_.find(node) != replicated_nodes_map_.end()) { in ReplicateNode()51 replicated_nodes_map_.emplace(node, std::move(replicated_nodes)); in ReplicateNode()61 replicated_nodes_map_.at(edge->dst()); in ReplicateFromRegularDeviceToCompositeDevice()72 replicated_nodes_map_.at(edge->src()); in ReplicateFromCompositeDeviceToCompositeDevice()74 replicated_nodes_map_.at(edge->dst()); in ReplicateFromCompositeDeviceToCompositeDevice()97 replicated_nodes_map_.at(edge->src()); in ReplicateFromCompositeDeviceToRegularDevice()163 for (const auto& entry : replicated_nodes_map_) { in RemoveDeadReplicatedArgs()174 absl::flat_hash_map<const Node*, std::vector<Node*>> replicated_nodes_map_; member in tensorflow::__anone6d961b90111::ReplicateHelper