Searched refs:DependencyHloOrdering (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_ordering.h | 149 class DependencyHloOrdering : public PredecessorHloOrdering { 151 explicit DependencyHloOrdering(const HloModule* module); 152 ~DependencyHloOrdering() override = default;
|
D | buffer_liveness_test.cc | 125 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 221 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 255 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 381 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 417 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 458 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 523 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 585 module.get(), absl::make_unique<DependencyHloOrdering>(module.get())) in TEST_F() 678 absl::make_unique<DependencyHloOrdering>(module.get())) in Run() 691 auto hlo_ordering = absl::make_unique<DependencyHloOrdering>(module.get()); in RunWithHloDataflowAnalysis() [all …]
|
D | hlo_ordering_test.cc | 86 DependencyHloOrdering ordering(module.get()); in TEST_F() 154 DependencyHloOrdering ordering(module.get()); in TEST_F() 190 DependencyHloOrdering ordering(module.get()); in TEST_F() 241 DependencyHloOrdering ordering(module.get()); in TEST_F() 335 DependencyHloOrdering ordering(module.get()); in TEST_F() 374 DependencyHloOrdering ordering(module.get()); in TEST_F()
|
D | hlo_ordering.cc | 335 DependencyHloOrdering::DependencyHloOrdering(const HloModule* module) in DependencyHloOrdering() function in xla::DependencyHloOrdering 345 string DependencyHloOrdering::ToString() const { in ToString()
|
D | hlo_alias_analysis_test.cc | 94 DependencyHloOrdering ordering(module_.get()); in AnyValuesInSameBufferInterfere() 1040 DependencyHloOrdering ordering(module_.get()); in TEST_F() 1083 DependencyHloOrdering ordering(module_.get()); in TEST_F()
|
D | copy_insertion.cc | 1183 DependencyHloOrdering dep_ordering(module); in Run() 1197 VerifyNoLiveRangeInterference(DependencyHloOrdering(module), module)); in Run()
|
D | buffer_assignment_test.cc | 92 module, absl::make_unique<DependencyHloOrdering>(module), in RunBufferAssignment() 103 module, absl::make_unique<DependencyHloOrdering>(module), in RunBufferAssignmentNoBuffersForConstants() 119 module, absl::make_unique<DependencyHloOrdering>(module), in RunBufferAssignmentNoBuffersReuseForAdd() 132 module, absl::make_unique<DependencyHloOrdering>(module), in RunColoredBufferAssignment()
|
D | hlo_dataflow_analysis_test.cc | 1248 DependencyHloOrdering ordering(module_.get()); in TEST_P() 1436 DependencyHloOrdering ordering(module_.get()); in TEST_P() 1468 DependencyHloOrdering ordering(module_.get()); in TEST_P() 1567 DependencyHloOrdering ordering(module_.get()); in TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_noalias_test.cc | 65 absl::make_unique<DependencyHloOrdering>(hlo_module.get()), in TEST_F()
|