Home
last modified time | relevance | path

Searched refs:thunk_schedule_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc64 thunk_schedule_(std::move(params.thunk_schedule)), in GpuExecutable()
148 sub_streams.reserve(thunk_schedule_->StreamCount() - 1); in ExecuteThunks()
149 while (sub_streams.size() + 1 < thunk_schedule_->StreamCount()) { in ExecuteThunks()
168 for (Thunk* thunk : thunk_schedule_->TotalOrder()) { in ExecuteThunks()
174 int32 stream_no = thunk_schedule_->StreamNumberForThunk(thunk); in ExecuteThunks()
178 for (const Thunk* dependency : thunk_schedule_->DependsOn(thunk)) { in ExecuteThunks()
200 if (thunk_schedule_->Depended(thunk)) { in ExecuteThunks()
549 for (Thunk* thunk : thunk_schedule_->TotalOrder()) { in ExecuteAsyncOnStreamImpl()
Dgpu_executable.h194 const std::unique_ptr<const ThunkSchedule> thunk_schedule_; variable