Searched refs:computations_ (Results 1 – 3 of 3) sorted by relevance
124 return {MakeUnwrappingIterator(computations_.begin()), in computations()125 MakeUnwrappingIterator(computations_.end())}; in computations()130 return {MakeUnwrappingIterator(computations_.begin()), in computations()131 MakeUnwrappingIterator(computations_.end())}; in computations()135 int64 computation_count() const { return computations_.size(); } in computation_count()221 std::vector<std::unique_ptr<HloComputation>> computations_; variable
87 computations_.push_back(std::move(computation)); in AddComputationInternal()88 return computations_.back().get(); in AddComputationInternal()99 std::find_if(computations_.begin(), computations_.end(), in RemoveEmbeddedComputation()104 computations_.erase(it); in RemoveEmbeddedComputation()119 new_computations.reserve(computations_.size()); in ReplaceComputations()121 for (std::unique_ptr<HloComputation>& computation : computations_) { in ReplaceComputations()190 computations_ = std::move(new_computations); in ReplaceComputations()481 for (const auto& computation : computations_) { in instruction_count()492 for (auto& computation : computations_) { in MakeComputationPostOrder()506 for (auto& computation : computations_) { in MakeComputationPostOrder()[all …]
233 std::vector<std::unique_ptr<HloComputation>> computations_; member in xla::tools::__anon9449f1f20111::HloParser289 for (int i = 0; i < computations_.size(); i++) { in ParseComputations()295 computations_[i].get() != entry_computation) || in ParseComputations()296 (entry_computation == nullptr && i != computations_.size() - 1)) { in ParseComputations()297 module_->AddEmbeddedComputation(std::move(computations_[i])); in ParseComputations()301 module_->AddEntryComputation(std::move(computations_[i])); in ParseComputations()359 computations_.emplace_back(builder->Build(root)); in ParseComputation()360 HloComputation* computation = computations_.back().get(); in ParseComputation()