Searched refs:computations_ (Results 1 – 4 of 4) sorted by relevance
51 computations_[old_computation] = new_computation; in MapComputation()63 return FindOrDefault(computations_, old_computation, nullptr); in FindComputation()73 return FindOrDie(computations_, old_computation); in GetComputation()83 return computations_; in cloned_computations()90 absl::flat_hash_map<const HloComputation*, HloComputation*> computations_; variable
168 return {MakeUnwrappingIterator(computations_.begin()), in computations()169 MakeUnwrappingIterator(computations_.end())}; in computations()174 return {MakeUnwrappingIterator(computations_.begin()), in computations()175 MakeUnwrappingIterator(computations_.end())}; in computations()183 int64 computation_count() const { return computations_.size(); } in computation_count()187 CHECK(idx >= 0 && idx < computations_.size()); in mutable_computation()188 return computations_[idx].get(); in mutable_computation()196 for (auto& comp : computations_) { in Cleanup()390 std::vector<std::unique_ptr<HloComputation>> computations_; variable
118 computations_.push_back(std::move(computation)); in AddComputationInternal()119 return computations_.back().get(); in AddComputationInternal()142 computations_, [&to_remove](const std::unique_ptr<HloComputation>& comp) { in RemoveEmbeddedComputation()145 TF_RET_CHECK(it != computations_.end()); in RemoveEmbeddedComputation()147 computations_.erase(it); in RemoveEmbeddedComputation()163 new_computations.reserve(computations_.size()); in ReplaceComputations()165 for (std::unique_ptr<HloComputation>& computation : computations_) { in ReplaceComputations()232 computations_ = std::move(new_computations); in ReplaceComputations()594 for (const auto& computation : computations_) { in instruction_count()621 for (auto& computation : computations_) { in MakeComputationPostOrder()[all …]
536 std::vector<std::unique_ptr<HloComputation>> computations_; member in xla::__anonf806807d0111::HloParserImpl842 for (int i = 0; i < computations_.size(); i++) { in ParseComputations()848 computations_[i].get() != entry_computation) || in ParseComputations()849 (entry_computation == nullptr && i != computations_.size() - 1)) { in ParseComputations()850 module->AddEmbeddedComputation(std::move(computations_[i])); in ParseComputations()853 auto computation = module->AddEntryComputation(std::move(computations_[i])); in ParseComputations()952 computations_.emplace_back(builder.Build(root)); in ParseInstructionList()953 *computation = computations_.back().get(); in ParseInstructionList()5236 for (auto& comp : computations_) { in ParseSingleInstruction()