Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.cc349 SequentialHloOrdering::SequentialHloOrdering(const HloSchedule& schedule) in SequentialHloOrdering() function in xla::SequentialHloOrdering
354 SequentialHloOrdering::SequentialHloOrdering(HloSchedule&& schedule) in SequentialHloOrdering() function in xla::SequentialHloOrdering
359 void SequentialHloOrdering::Initialize() { in Initialize()
370 bool SequentialHloOrdering::ExecutesBeforeInSameComputation( in ExecutesBeforeInSameComputation()
380 const HloInstructionSequence* SequentialHloOrdering::SequentialOrder( in SequentialOrder()
387 string SequentialHloOrdering::ToString() const { in ToString()
Dhlo_ordering.h182 class SequentialHloOrdering : public HloOrdering {
184 SequentialHloOrdering(const HloSchedule& schedule);
185 SequentialHloOrdering(HloSchedule&& schedule);
186 ~SequentialHloOrdering() override = default;
Dhlo_memory_scheduler_test.cc89 SequentialHloOrdering ordering(module->schedule()); in TEST_F()
145 SequentialHloOrdering ordering(schedule); in TEST_F()
189 SequentialHloOrdering ordering(schedule); in TEST_F()
239 SequentialHloOrdering ordering(schedule); in TEST_F()
Dhlo_ordering_test.cc426 SequentialHloOrdering ordering(schedule); in TEST_F()
482 SequentialHloOrdering ordering(schedule); in TEST_F()
Dbuffer_liveness_test.cc174 absl::make_unique<SequentialHloOrdering>(schedule)) in TEST_F()
299 absl::make_unique<SequentialHloOrdering>(schedule)) in TEST_F()
348 absl::make_unique<SequentialHloOrdering>(schedule)) in TEST_F()
Dhlo_alias_analysis_test.cc1095 SequentialHloOrdering ordering(schedule); in TEST_F()
1102 SequentialHloOrdering ordering(schedule); in TEST_F()
Dbuffer_assignment_test.cc146 module, absl::make_unique<SequentialHloOrdering>(schedule), in RunBufferAssignmentWithInstructionSequence()
1893 module, absl::make_unique<SequentialHloOrdering>(schedule), in RunBufferAssignment()
2213 module.get(), absl::make_unique<SequentialHloOrdering>(schedule), in TEST_F()
2447 module.get(), absl::make_unique<SequentialHloOrdering>(schedule), in TEST_F()
Dhlo_dataflow_analysis_test.cc1289 SequentialHloOrdering ordering(schedule); in TEST_P()
1375 SequentialHloOrdering ordering(schedule); in TEST_P()
1511 SequentialHloOrdering ordering(schedule); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc646 absl::make_unique<SequentialHloOrdering>(schedule), in RunBackend()
831 absl::make_unique<SequentialHloOrdering>(schedule), in CompileAheadOfTime()