Searched refs:GetInflightTaskCount (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | run_handler.cc | 283 int64_t ThreadWorkSource::GetInflightTaskCount(bool is_blocking) { in GetInflightTaskCount() function in tensorflow::internal::ThreadWorkSource 308 ", inter inflight = ", GetInflightTaskCount(true), in ToString() 310 ", intra inflight = ", GetInflightTaskCount(false)); in ToString() 509 (*tws)->GetInflightTaskCount(true) < max_blocking_inflight) { in FindTask() 598 tws->GetInflightTaskCount(true) < kMaxBlockingInflight) { in WorkerLoop() 715 if (tws->GetInflightTaskCount(true) >= max_blocking_inflight) { in WaitForWork()
|
H A D | run_handler.h | 179 int64_t GetInflightTaskCount(bool is_blocking);
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/ |
H A D | run_handler.cc | 291 int64_t ThreadWorkSource::GetInflightTaskCount(bool is_blocking) { in GetInflightTaskCount() function in tfrt::tf::internal::ThreadWorkSource 330 ", inter inflight = ", GetInflightTaskCount(true), in ToString() 332 ", intra inflight = ", GetInflightTaskCount(false)); in ToString() 492 (*tws)->GetInflightTaskCount(true) < max_blocking_inflight) { in FindTask()
|
H A D | run_handler.h | 237 int64_t GetInflightTaskCount(bool is_blocking);
|