Searched refs:computations_ (Results 1 – 4 of 4) sorted by relevance
153 return {MakeUnwrappingIterator(computations_.begin()), in computations()154 MakeUnwrappingIterator(computations_.end())}; in computations()159 return {MakeUnwrappingIterator(computations_.begin()), in computations()160 MakeUnwrappingIterator(computations_.end())}; in computations()168 int64 computation_count() const { return computations_.size(); } in computation_count()172 CHECK(idx >= 0 && idx < computations_.size()); in mutable_computation()173 return computations_[idx].get(); in mutable_computation()294 std::vector<std::unique_ptr<HloComputation>> computations_; variable
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
99 computations_.push_back(std::move(computation)); in AddComputationInternal()100 return computations_.back().get(); in AddComputationInternal()111 computations_, [&to_remove](const std::unique_ptr<HloComputation>& comp) { in RemoveEmbeddedComputation()115 computations_.erase(it); in RemoveEmbeddedComputation()130 new_computations.reserve(computations_.size()); in ReplaceComputations()132 for (std::unique_ptr<HloComputation>& computation : computations_) { in ReplaceComputations()197 computations_ = std::move(new_computations); in ReplaceComputations()493 for (const auto& computation : computations_) { in instruction_count()504 for (auto& computation : computations_) { in MakeComputationPostOrder()518 for (auto& computation : computations_) { in MakeComputationPostOrder()[all …]
366 std::vector<std::unique_ptr<HloComputation>> computations_; member in xla::__anonc071bf1f0111::HloParser519 for (int i = 0; i < computations_.size(); i++) { in ParseComputations()525 computations_[i].get() != entry_computation) || in ParseComputations()526 (entry_computation == nullptr && i != computations_.size() - 1)) { in ParseComputations()527 module->AddEmbeddedComputation(std::move(computations_[i])); in ParseComputations()530 auto computation = module->AddEntryComputation(std::move(computations_[i])); in ParseComputations()628 computations_.emplace_back(builder.Build(root)); in ParseInstructionList()629 *computation = computations_.back().get(); in ParseInstructionList()4090 for (auto& comp : computations_) { in ParseSingleInstruction()