| /external/tensorflow/tensorflow/compiler/xla/service/ |
| buffer_liveness.h | 26 #include "tensorflow/compiler/xla/service/hlo_ordering.h"
43 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering);
66 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } function in class:xla::BufferLiveness
86 std::unique_ptr<HloOrdering> hlo_ordering)
87 : module_(module), hlo_ordering_(std::move(hlo_ordering)) {}
|
| buffer_liveness.cc | 39 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering) {
41 new BufferLiveness(module, std::move(hlo_ordering)));
|
| buffer_assignment.cc | 637 liveness_->hlo_ordering().SequentialOrder(*computation);
736 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
743 return assigner.CreateAssignment(module, std::move(hlo_ordering),
757 assignment->liveness().hlo_ordering().call_graph();
1154 const HloOrdering& hlo_ordering = assignment->liveness().hlo_ordering(); local
[all...] |
| buffer_assignment.h | 549 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
568 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
587 // assignment->liveness().hlo_ordering().SequentialOrder. If
|
| buffer_liveness_test.cc | 691 auto hlo_ordering = absl::make_unique<DependencyHloOrdering>(module.get()); local
696 return hlo_ordering->MayInterfere(
[all...] |
