Home
last modified time | relevance | path

Searched refs:thread_local_computations (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment.cc106 std::vector<const HloComputation*>* thread_local_computations, in GatherComputationsByAllocationType() argument
202 thread_local_computations->push_back(computation); in GatherComputationsByAllocationType()
1756 std::vector<const HloComputation*> thread_local_computations; in CreateAssignment() local
1759 module, &thread_local_computations, &global_computations)); in CreateAssignment()
1789 for (auto* computation : thread_local_computations) { in CreateAssignment()
H A Dbuffer_assignment.h53 std::vector<const HloComputation*>* thread_local_computations,