Searched refs:BufferLiveness (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | buffer_liveness.cc | 38 StatusOr<std::unique_ptr<BufferLiveness>> BufferLiveness::Run( in Run() 40 std::unique_ptr<BufferLiveness> liveness( in Run() 41 new BufferLiveness(module, std::move(hlo_ordering))); in Run() 46 Status BufferLiveness::Analyze() { in Analyze() 76 string BufferLiveness::ToString() const { in ToString() 93 bool BufferLiveness::live_range_strictly_before(const LogicalBuffer& a, in live_range_strictly_before() 148 bool BufferLiveness::MayInterfere(const LogicalBuffer& a, in MayInterfere() 166 bool BufferLiveness::MaybeLiveOut(const LogicalBuffer& buffer) const { in MaybeLiveOut()
|
D | buffer_liveness.h | 36 class BufferLiveness { 38 using Colorer = std::function<Status(const BufferLiveness& buffer_liveness)>; 42 static StatusOr<std::unique_ptr<BufferLiveness>> Run( 73 return [](const BufferLiveness& buffer_liveness) { in DefaultColorer() 85 explicit BufferLiveness(const HloModule* module, in BufferLiveness() function
|
D | buffer_liveness_test.cc | 39 const LogicalBuffer& GetBuffer(const BufferLiveness& liveness, in GetBuffer() 53 bool InstructionsMayInterfere(const BufferLiveness& liveness, in InstructionsMayInterfere() 65 bool TupleElementsMayInterfere(const BufferLiveness& liveness, in TupleElementsMayInterfere() 90 bool InstructionMaybeLiveOut(const BufferLiveness& liveness, in InstructionMaybeLiveOut() 124 BufferLiveness::Run( in TEST_F() 173 BufferLiveness::Run(module.get(), in TEST_F() 220 BufferLiveness::Run( in TEST_F() 254 BufferLiveness::Run( in TEST_F() 298 BufferLiveness::Run(module.get(), in TEST_F() 347 BufferLiveness::Run(module.get(), in TEST_F() [all …]
|
D | buffer_assignment.h | 443 const BufferLiveness& liveness() const { return *liveness_; } in liveness() 474 std::unique_ptr<BufferLiveness> liveness, in BufferAssignment() 522 const std::unique_ptr<BufferLiveness> liveness_; 554 BufferLiveness::Colorer colorer = BufferLiveness::DefaultColorer(), 559 BufferLiveness::Colorer colorer, in BufferAssigner() 618 const HloModule* module, const BufferLiveness& buffer_liveness, 641 const BufferLiveness& buffer_liveness, 655 BufferLiveness::Colorer colorer_;
|
D | buffer_assignment.cc | 740 BufferLiveness::Colorer colorer, ReuseAllocationFunction reuse_checker) { in Run() 928 const BufferLiveness& liveness = assignment->liveness(); in AssignBuffersForComputation() 1419 const BufferLiveness& buffer_liveness, in MergeColocatedBufferSets() 1531 const HloModule* module, const BufferLiveness& buffer_liveness, in BuildColocatedBufferSets() 1758 TF_ASSIGN_OR_RETURN(std::unique_ptr<BufferLiveness> liveness, in CreateAssignment() 1759 BufferLiveness::Run(module, std::move(hlo_ordering))); in CreateAssignment()
|
D | buffer_assignment_test.cc | 124 /*colorer=*/BufferLiveness::DefaultColorer(), in RunBufferAssignmentNoBuffersReuseForAdd() 130 HloModule* module, BufferLiveness::Colorer colorer, int64 alignment = 1) { in RunColoredBufferAssignment() 579 auto colorer = [](const BufferLiveness& buffer_liveness) { in TEST_F() 640 auto colorer = [](const BufferLiveness& buffer_liveness) { in TEST_F()
|