Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.cc35 bool HloOrdering::ExecutesBefore(const HloInstruction* a, in ExecutesBefore()
50 HloOrdering::ExecutionConstraint HloOrdering::GetExecutionConstraint( in GetExecutionConstraint()
143 bool HloOrdering::IsDefinedBefore(const HloValue& a, const HloValue& b) const { in IsDefinedBefore()
212 bool HloOrdering::UsesBeforeValueDefinition( in UsesBeforeValueDefinition()
222 case HloOrdering::ExecutionConstraint::kIsSame: in UsesBeforeValueDefinition()
235 case HloOrdering::ExecutionConstraint::kRunExclusiveAfter: in UsesBeforeValueDefinition()
256 case HloOrdering::ExecutionConstraint::kRunExclusiveBefore: in UsesBeforeValueDefinition()
257 case HloOrdering::ExecutionConstraint::kRunBefore: in UsesBeforeValueDefinition()
261 case HloOrdering::ExecutionConstraint::kRunAfter: in UsesBeforeValueDefinition()
262 case HloOrdering::ExecutionConstraint::kUnordered: in UsesBeforeValueDefinition()
[all …]
Dhlo_ordering.h38 class HloOrdering {
40 explicit HloOrdering(const HloModule* module) in HloOrdering() function
42 virtual ~HloOrdering() = default;
61 HloOrdering::ExecutionConstraint GetExecutionConstraint(
115 class PredecessorHloOrdering : public HloOrdering {
203 class SequentialHloOrdering : public HloOrdering {
Dbuffer_assignment.h463 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } in hlo_ordering()
498 std::unique_ptr<HloOrdering> hlo_ordering, in BufferAssignment()
567 const std::unique_ptr<HloOrdering> hlo_ordering_;
587 using Colorer = std::function<Status(HloAliasAnalysis*, const HloOrdering&)>;
590 return [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in DefaultColorer()
613 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
634 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
Dcopy_insertion.h66 Status RemoveUnnecessaryCopies(const HloOrdering& ordering,
Dhlo_alias_analysis.h102 bool HasLiveRangeInterference(const HloOrdering& ordering) const;
Dcopy_insertion.cc488 const HloOrdering& ordering) in CopyRemover()
959 const HloOrdering& ordering_;
1177 Status CopyInsertion::RemoveUnnecessaryCopies(const HloOrdering& ordering, in RemoveUnnecessaryCopies()
Dbuffer_assignment.cc910 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in Run()
1414 const HloOrdering& hlo_ordering = assignment->hlo_ordering(); in AssignBuffersWithSequentialOrdering()
1626 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in CreateAssignment()
Dhlo_alias_analysis.cc631 const HloOrdering& ordering) const { in HasLiveRangeInterference()
Dbuffer_assignment_test.cc585 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
651 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
Dhlo_dataflow_analysis_test.cc76 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_;