Home
last modified time | relevance | path

Searched refs:hlo_ordering_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_liveness.cc81 pieces.push_back(hlo_ordering_->ToString()); in ToString()
98 if (!hlo_ordering_->ExecutesBefore(a.instruction(), b.instruction())) { in live_range_strictly_before()
110 !hlo_ordering_->ExecutesBefore(user, b.instruction())) { in live_range_strictly_before()
155 hlo_ordering_->ExecutesBefore(b_instruction, a_instruction)) { in MayInterfere()
159 hlo_ordering_->ExecutesBefore(a_instruction, b_instruction)) { in MayInterfere()
Dbuffer_liveness.h66 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } in hlo_ordering()
87 : module_(module), hlo_ordering_(std::move(hlo_ordering)) {} in BufferLiveness()
99 std::unique_ptr<HloOrdering> hlo_ordering_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.h57 return std::move(hlo_ordering_); in ConsumeHloOrdering()
64 std::unique_ptr<HloOrdering> hlo_ordering_; variable
Dgpu_hlo_schedule.cc214 schedule->hlo_ordering_ = absl::make_unique<GpuHloOrdering>( in Build()