Searched refs:sequences_ (Results 1 – 4 of 4) sorted by relevance
71 for (const auto& id_sequence : sequences_) { in ToProto()92 sequences_[computation->unique_id()] = std::move(sequence); in set_sequence()97 auto it = sequences_.find(computation->unique_id()); in GetOrCreateSequence()98 if (it == sequences_.end()) { in GetOrCreateSequence()101 return sequences_[computation->unique_id()]; in GetOrCreateSequence()109 return sequences_.at(computation->unique_id()); in sequence()123 for (int id : sequences_.at(computation->unique_id()).ids()) { in UpdateComputationSchedule()185 for (int id : sequences_.at(computation->unique_id()).ids()) { in UpdateComputationSchedule()207 TF_RET_CHECK(sequences_.contains(computation->unique_id())) in Update()210 if (sequences_.size() > nonfusion_computations.size()) { in Update()[all …]
108 return sequences_; in sequences()113 return sequences_.contains(computation->unique_id()); in is_computation_scheduled()138 bool empty() const { return sequences_.empty(); } in empty()151 absl::flat_hash_map<int64, HloInstructionSequence> sequences_; variable
166 for (uint32_t i = 0; i < sequences_.size(); ++i) { in ~BasicPortAllocatorSession()169 sequences_[i]->Clear(); in ~BasicPortAllocatorSession()179 for (uint32_t i = 0; i < sequences_.size(); ++i) in ~BasicPortAllocatorSession()180 delete sequences_[i]; in ~BasicPortAllocatorSession()207 for (uint32_t i = 0; i < sequences_.size(); ++i) in ClearGettingPorts()208 sequences_[i]->Stop(); in ClearGettingPorts()287 for (std::vector<AllocationSequence*>::iterator it = sequences_.begin(); in OnConfigStop()288 it != sequences_.end() && !send_signal; ++it) { in OnConfigStop()395 sequences_.push_back(sequence); in DoAllocate()406 for (AllocationSequence* sequence : sequences_) { in OnNetworksChanged()[all …]
195 std::vector<AllocationSequence*> sequences_; variable