Home
last modified time | relevance | path

Searched refs:BufferLiveness (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_liveness.cc38 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()
Dbuffer_liveness.h36 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
Dbuffer_liveness_test.cc39 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 …]
Dbuffer_assignment.h443 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_;
Dbuffer_assignment.cc740 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()
Dbuffer_assignment_test.cc124 /*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()