Searched refs:global_computations (Results 1 – 2 of 2) sorted by relevance
129 std::vector<const HloComputation*>* global_computations) { in GatherComputationsByAllocationType() argument219 global_computations->push_back(computation); in GatherComputationsByAllocationType()1669 std::vector<const HloComputation*> global_computations; in CreateAssignment() local1671 module, &thread_local_computations, &global_computations)); in CreateAssignment()1679 global_computations, in CreateAssignment()1686 buffers_to_assign_sequentially.size() == global_computations.size(); in CreateAssignment()
56 std::vector<const HloComputation*>* global_computations);