Searched refs:thread_local_computations (Results 1 – 2 of 2) sorted by relevance
106 std::vector<const HloComputation*>* thread_local_computations, in GatherComputationsByAllocationType() argument202 thread_local_computations->push_back(computation); in GatherComputationsByAllocationType()1756 std::vector<const HloComputation*> thread_local_computations; in CreateAssignment() local1759 module, &thread_local_computations, &global_computations)); in CreateAssignment()1789 for (auto* computation : thread_local_computations) { in CreateAssignment()
53 std::vector<const HloComputation*>* thread_local_computations,