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.cc122 std::vector<const HloComputation*>* thread_local_computations, in GatherComputationsByAllocationType() argument
211 thread_local_computations->push_back(computation); in GatherComputationsByAllocationType()
1788 std::vector<const HloComputation*> thread_local_computations; in CreateAssignment() local
1791 module, &thread_local_computations, &global_computations)); in CreateAssignment()
1814 for (auto* computation : thread_local_computations) { in CreateAssignment()
Dbuffer_assignment.h52 std::vector<const HloComputation*>* thread_local_computations,