Home
last modified time | relevance | path

Searched refs:HloOrdering (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.h38 class HloOrdering {
40 HloOrdering(const HloModule* module) in HloOrdering() function
42 virtual ~HloOrdering() = default;
94 class PredecessorHloOrdering : public HloOrdering {
182 class SequentialHloOrdering : public HloOrdering {
Dbuffer_liveness.h43 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_;
Dhlo_ordering.cc35 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()
Dcopy_insertion.h77 Status RemoveUnnecessaryCopies(const HloOrdering& ordering,
94 Status VerifyNoLiveRangeInterference(const HloOrdering& ordering,
Dhlo_alias_analysis.h100 bool HasLiveRangeInterference(const HloOrdering& ordering) const;
Dcopy_insertion.cc476 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()
Dbuffer_liveness.cc39 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering) { in Run()
Dbuffer_assignment.h549 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
568 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
Dbuffer_assignment.cc736 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()
Dhlo_alias_analysis.cc531 const HloOrdering& ordering) const { in HasLiveRangeInterference()
Dhlo_dataflow_analysis_test.cc73 bool InstructionsMayInterfere(const HloOrdering& ordering, in InstructionsMayInterfere()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.h56 std::unique_ptr<HloOrdering> ConsumeHloOrdering() { in ConsumeHloOrdering()
64 std::unique_ptr<HloOrdering> hlo_ordering_;