Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.h149 class DependencyHloOrdering : public PredecessorHloOrdering {
151 explicit DependencyHloOrdering(const HloModule* module);
152 ~DependencyHloOrdering() override = default;
Dbuffer_liveness_test.cc125 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 …]
Dhlo_ordering_test.cc86 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()
Dhlo_ordering.cc335 DependencyHloOrdering::DependencyHloOrdering(const HloModule* module) in DependencyHloOrdering() function in xla::DependencyHloOrdering
345 string DependencyHloOrdering::ToString() const { in ToString()
Dhlo_alias_analysis_test.cc94 DependencyHloOrdering ordering(module_.get()); in AnyValuesInSameBufferInterfere()
1040 DependencyHloOrdering ordering(module_.get()); in TEST_F()
1083 DependencyHloOrdering ordering(module_.get()); in TEST_F()
Dcopy_insertion.cc1183 DependencyHloOrdering dep_ordering(module); in Run()
1197 VerifyNoLiveRangeInterference(DependencyHloOrdering(module), module)); in Run()
Dbuffer_assignment_test.cc92 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()
Dhlo_dataflow_analysis_test.cc1248 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/
Dcpu_noalias_test.cc65 absl::make_unique<DependencyHloOrdering>(hlo_module.get()), in TEST_F()