Searched refs:hlo_ordering (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_liveness.h | 43 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering);
66 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } in hlo_ordering() function
86 std::unique_ptr<HloOrdering> hlo_ordering) in BufferLiveness() argument
87 : module_(module), hlo_ordering_(std::move(hlo_ordering)) {} in BufferLiveness()
|
| D | buffer_liveness.cc | 39 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering) { in Run() argument
41 new BufferLiveness(module, std::move(hlo_ordering))); in Run()
|
| D | buffer_assignment.cc | 637 liveness_->hlo_ordering().SequentialOrder(*computation); in ComputeSummaryStats()
736 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in Run() argument
743 return assigner.CreateAssignment(module, std::move(hlo_ordering), in Run()
757 assignment->liveness().hlo_ordering().call_graph(); in MayInterfereAcrossSubcomputations()
930 liveness.hlo_ordering().SequentialOrder(*computation) != nullptr; in AssignBuffersForComputation()
1154 const HloOrdering& hlo_ordering = assignment->liveness().hlo_ordering(); in AssignBuffersWithSequentialOrdering() local
1180 hlo_ordering.SequentialOrder(*computation); in AssignBuffersWithSequentialOrdering()
1215 hlo_ordering.SequentialOrder(*computation); in AssignBuffersWithSequentialOrdering()
1755 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in CreateAssignment() argument
1759 BufferLiveness::Run(module, std::move(hlo_ordering))); in CreateAssignment()
|
| D | BUILD | 389 ":hlo_ordering",
1079 ":hlo_ordering",
1156 ":hlo_ordering",
1175 name = "hlo_ordering",
1176 srcs = ["hlo_ordering.cc"],
1177 hdrs = ["hlo_ordering.h"],
1205 ":hlo_ordering",
1224 ":hlo_ordering",
1242 ":hlo_ordering",
1338 ":hlo_ordering",
[all …]
|
| D | buffer_assignment.h | 549 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
568 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
|
| D | buffer_liveness_test.cc | 691 auto hlo_ordering = absl::make_unique<DependencyHloOrdering>(module.get()); in RunWithHloDataflowAnalysis() local
696 return hlo_ordering->MayInterfere( in RunWithHloDataflowAnalysis()
|
| D | hlo.proto | 451 reserved "hlo_ordering";
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 127 "//tensorflow/compiler/xla/service:hlo_ordering",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 987 "//tensorflow/compiler/xla/service:hlo_ordering",
|