Searched refs:global_computations (Results 1 – 2 of 2) sorted by relevance
123 std::vector<const HloComputation*>* global_computations) { in GatherComputationsByAllocationType() argument213 global_computations->push_back(computation); in GatherComputationsByAllocationType()1789 std::vector<const HloComputation*> global_computations; in CreateAssignment() local1791 module, &thread_local_computations, &global_computations)); in CreateAssignment()1797 for (auto* computation : global_computations) { in CreateAssignment()1807 buffers_to_assign_sequentially.size() == global_computations.size(); in CreateAssignment()
53 std::vector<const HloComputation*>* global_computations);