Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc123 std::vector<const HloComputation*>* global_computations) { in GatherComputationsByAllocationType() argument
213 global_computations->push_back(computation); in GatherComputationsByAllocationType()
1789 std::vector<const HloComputation*> global_computations; in CreateAssignment() local
1791 module, &thread_local_computations, &global_computations)); in CreateAssignment()
1797 for (auto* computation : global_computations) { in CreateAssignment()
1807 buffers_to_assign_sequentially.size() == global_computations.size(); in CreateAssignment()
Dbuffer_assignment.h53 std::vector<const HloComputation*>* global_computations);