Searched refs:SequentialHloOrdering (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_ordering.cc | 349 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()
|
D | hlo_ordering.h | 182 class SequentialHloOrdering : public HloOrdering { 184 SequentialHloOrdering(const HloSchedule& schedule); 185 SequentialHloOrdering(HloSchedule&& schedule); 186 ~SequentialHloOrdering() override = default;
|
D | hlo_memory_scheduler_test.cc | 89 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()
|
D | hlo_ordering_test.cc | 426 SequentialHloOrdering ordering(schedule); in TEST_F() 482 SequentialHloOrdering ordering(schedule); in TEST_F()
|
D | buffer_liveness_test.cc | 174 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()
|
D | hlo_alias_analysis_test.cc | 1095 SequentialHloOrdering ordering(schedule); in TEST_F() 1102 SequentialHloOrdering ordering(schedule); in TEST_F()
|
D | buffer_assignment_test.cc | 146 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()
|
D | hlo_dataflow_analysis_test.cc | 1289 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/ |
D | cpu_compiler.cc | 646 absl::make_unique<SequentialHloOrdering>(schedule), in RunBackend() 831 absl::make_unique<SequentialHloOrdering>(schedule), in CompileAheadOfTime()
|