Home
last modified time | relevance | path

Searched refs:SequentialHloOrdering (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.h203 class SequentialHloOrdering : public HloOrdering {
205 explicit SequentialHloOrdering(const HloSchedule& schedule);
206 explicit SequentialHloOrdering(HloSchedule&& schedule);
207 ~SequentialHloOrdering() override = default;
Dhlo_ordering.cc486 SequentialHloOrdering::SequentialHloOrdering(const HloSchedule& schedule) in SequentialHloOrdering() function in xla::SequentialHloOrdering
491 SequentialHloOrdering::SequentialHloOrdering(HloSchedule&& schedule) in SequentialHloOrdering() function in xla::SequentialHloOrdering
496 void SequentialHloOrdering::Initialize() { in Initialize()
507 bool SequentialHloOrdering::ExecutesBeforeInSameComputation( in ExecutesBeforeInSameComputation()
517 const HloInstructionSequence* SequentialHloOrdering::SequentialOrder( in SequentialOrder()
524 string SequentialHloOrdering::ToString() const { in ToString()
Dhlo_memory_scheduler_test.cc108 SequentialHloOrdering ordering(module->schedule()); in TEST_F()
168 SequentialHloOrdering ordering(schedule); in TEST_F()
212 SequentialHloOrdering ordering(schedule); in TEST_F()
255 SequentialHloOrdering ordering(schedule); in TEST_F()
306 SequentialHloOrdering ordering(schedule); in TEST_F()
Dhlo_ordering_test.cc432 SequentialHloOrdering ordering(schedule); in TEST_F()
488 SequentialHloOrdering ordering(schedule); in TEST_F()
Dhlo_alias_analysis_test.cc1247 SequentialHloOrdering ordering(schedule); in TEST_F()
1254 SequentialHloOrdering ordering(schedule); in TEST_F()
Dbuffer_assignment_test.cc140 module, absl::make_unique<SequentialHloOrdering>(schedule), in RunBufferAssignmentWithInstructionSequence()
2042 module, absl::make_unique<SequentialHloOrdering>(schedule), in RunBufferAssignment()
2361 module.get(), absl::make_unique<SequentialHloOrdering>(schedule), in TEST_F()
2629 module.get(), absl::make_unique<SequentialHloOrdering>(schedule), in TEST_F()
Dhlo_dataflow_analysis_test.cc1397 SequentialHloOrdering ordering(schedule); in TEST_P()
1483 SequentialHloOrdering ordering(schedule); in TEST_P()
1619 SequentialHloOrdering ordering(schedule); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc610 absl::make_unique<SequentialHloOrdering>(schedule), in RunHloPassesAndBufferAssignement()
726 absl::make_unique<SequentialHloOrdering>(schedule), in RunBackend()
922 absl::make_unique<SequentialHloOrdering>(schedule), in CompileAheadOfTime()