Searched refs:HloOrdering (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_ordering.h | 38 class HloOrdering { 40 HloOrdering(const HloModule* module) in HloOrdering() function 42 virtual ~HloOrdering() = default; 94 class PredecessorHloOrdering : public HloOrdering { 182 class SequentialHloOrdering : public HloOrdering {
|
D | buffer_liveness.h | 43 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering); 66 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } in hlo_ordering() 86 std::unique_ptr<HloOrdering> hlo_ordering) in BufferLiveness() 99 std::unique_ptr<HloOrdering> hlo_ordering_;
|
D | hlo_ordering.cc | 35 bool HloOrdering::ExecutesBefore(const HloInstruction* a, in ExecutesBefore() 101 bool HloOrdering::IsDefinedBefore(const HloValue& a, const HloValue& b) const { in IsDefinedBefore() 170 bool HloOrdering::UseIsBeforeValueDefinition( in UseIsBeforeValueDefinition() 255 bool HloOrdering::LiveRangeStrictlyBefore( in LiveRangeStrictlyBefore() 297 bool HloOrdering::MayInterfere(const HloValue& a, const HloValue& b, in MayInterfere() 305 : HloOrdering(module) {} in PredecessorHloOrdering() 350 : HloOrdering(schedule.module()), schedule_(schedule) { in SequentialHloOrdering() 355 : HloOrdering(schedule.module()), schedule_(std::move(schedule)) { in SequentialHloOrdering()
|
D | copy_insertion.h | 77 Status RemoveUnnecessaryCopies(const HloOrdering& ordering, 94 Status VerifyNoLiveRangeInterference(const HloOrdering& ordering,
|
D | hlo_alias_analysis.h | 100 bool HasLiveRangeInterference(const HloOrdering& ordering) const;
|
D | copy_insertion.cc | 476 const HloOrdering& ordering) in CopyRemover() 941 const HloOrdering& ordering_; 1091 Status CopyInsertion::VerifyNoLiveRangeInterference(const HloOrdering& ordering, in VerifyNoLiveRangeInterference() 1099 Status CopyInsertion::RemoveUnnecessaryCopies(const HloOrdering& ordering, in RemoveUnnecessaryCopies()
|
D | buffer_liveness.cc | 39 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering) { in Run()
|
D | buffer_assignment.h | 549 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, 568 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
|
D | buffer_assignment.cc | 736 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in Run() 1154 const HloOrdering& hlo_ordering = assignment->liveness().hlo_ordering(); in AssignBuffersWithSequentialOrdering() 1755 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in CreateAssignment()
|
D | hlo_alias_analysis.cc | 531 const HloOrdering& ordering) const { in HasLiveRangeInterference()
|
D | hlo_dataflow_analysis_test.cc | 73 bool InstructionsMayInterfere(const HloOrdering& ordering, in InstructionsMayInterfere()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_hlo_schedule.h | 56 std::unique_ptr<HloOrdering> ConsumeHloOrdering() { in ConsumeHloOrdering() 64 std::unique_ptr<HloOrdering> hlo_ordering_;
|