Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc128 std::vector<const HloComputation*>* thread_local_computations, in GatherComputationsByAllocationType() argument
217 thread_local_computations->push_back(computation); in GatherComputationsByAllocationType()
1668 std::vector<const HloComputation*> thread_local_computations; in CreateAssignment() local
1671 module, &thread_local_computations, &global_computations)); in CreateAssignment()
1701 for (auto* computation : thread_local_computations) { in CreateAssignment()
Dbuffer_assignment.h55 std::vector<const HloComputation*>* thread_local_computations,