Home
last modified time | relevance | path

Searched refs:ThreadWorkSource (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler.cc127 ThreadWorkSource::ThreadWorkSource() in ThreadWorkSource() function in tfrt::tf::internal::ThreadWorkSource
145 ThreadWorkSource::~ThreadWorkSource() { in ~ThreadWorkSource()
151 Task ThreadWorkSource::EnqueueTask(Task t, bool is_blocking, in EnqueueTask()
231 Task ThreadWorkSource::PopBlockingTask() { in PopBlockingTask()
235 Task ThreadWorkSource::PopNonBlockingTask(int start_index, in PopNonBlockingTask()
252 int ThreadWorkSource::TaskQueueSize(bool is_blocking) { in TaskQueueSize()
264 int64_t ThreadWorkSource::GetTracemeId() { in GetTracemeId()
268 void ThreadWorkSource::SetTracemeId(int64_t value) { traceme_id_ = value; } in SetTracemeId()
270 void ThreadWorkSource::SetWaiter(uint64_t version, Waiter* waiter, in SetWaiter()
291 int64_t ThreadWorkSource::GetInflightTaskCount(bool is_blocking) { in GetInflightTaskCount()
[all …]
H A Drun_handler_test.cc162 internal::ThreadWorkSource tws; in TEST_P()
209 Eigen::MaxSizeVector<internal::ThreadWorkSource*> thread_work_sources(5); in TEST_P()
212 thread_work_sources[i] = new internal::ThreadWorkSource(); in TEST_P()
234 internal::ThreadWorkSource* tws; in TEST_P()
275 internal::ThreadWorkSource* tws; in TEST_P()
307 internal::ThreadWorkSource* tws; in TEST_P()
339 internal::ThreadWorkSource* tws; in TEST_P()
385 internal::ThreadWorkSource* tws; in TEST_P()
419 internal::ThreadWorkSource* tws; in TEST_P()
454 internal::ThreadWorkSource* tws; in TEST_P()
[all …]
H A Drun_handler.h217 class ThreadWorkSource {
219 ThreadWorkSource();
221 ~ThreadWorkSource();
336 void AddWorkToQueue(ThreadWorkSource* tws, bool is_blocking, TaskFunction fn);
341 const Eigen::MaxSizeVector<ThreadWorkSource*>& thread_work_sources);
362 const Eigen::MaxSizeVector<ThreadWorkSource*>& thread_work_sources,
363 bool* task_from_blocking_queue, ThreadWorkSource** tws);
376 std::unique_ptr<Eigen::MaxSizeVector<ThreadWorkSource*>>
381 std::unique_ptr<Eigen::MaxSizeVector<ThreadWorkSource*>>
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler.cc126 ThreadWorkSource::ThreadWorkSource() in ThreadWorkSource() function in tensorflow::internal::ThreadWorkSource
143 ThreadWorkSource::~ThreadWorkSource() { in ~ThreadWorkSource()
149 Task ThreadWorkSource::EnqueueTask(Task t, bool is_blocking) { in EnqueueTask()
219 Task ThreadWorkSource::PopBlockingTask() { in PopBlockingTask()
223 Task ThreadWorkSource::PopNonBlockingTask(int start_index, in PopNonBlockingTask()
240 void ThreadWorkSource::WaitForWork(int max_sleep_micros) { in WaitForWork()
245 int ThreadWorkSource::TaskQueueSize(bool is_blocking) { in TaskQueueSize()
257 int64_t ThreadWorkSource::GetTracemeId() { in GetTracemeId()
261 void ThreadWorkSource::SetTracemeId(int64_t value) { traceme_id_ = value; } in SetTracemeId()
263 void ThreadWorkSource::SetWaiter(uint64 version, Waiter* waiter, mutex* mutex) { in SetWaiter()
[all …]
H A Drun_handler.h157 class ThreadWorkSource {
159 ThreadWorkSource();
161 ~ThreadWorkSource();
235 void AddWorkToQueue(ThreadWorkSource* tws, bool is_blocking,
243 const Eigen::MaxSizeVector<ThreadWorkSource*>& thread_work_sources);
264 const Eigen::MaxSizeVector<ThreadWorkSource*>& thread_work_sources,
265 bool* task_from_blocking_queue, ThreadWorkSource** tws);
280 std::unique_ptr<Eigen::MaxSizeVector<ThreadWorkSource*>>
285 std::unique_ptr<Eigen::MaxSizeVector<ThreadWorkSource*>>
H A Drun_handler_test.cc125 internal::ThreadWorkSource tws; in TEST()
159 Eigen::MaxSizeVector<internal::ThreadWorkSource*> thread_work_sources(5); in TEST()
162 thread_work_sources[i] = new internal::ThreadWorkSource(); in TEST()
183 internal::ThreadWorkSource* tws; in TEST()
224 internal::ThreadWorkSource* tws; in TEST()
256 internal::ThreadWorkSource* tws; in TEST()
288 internal::ThreadWorkSource* tws; in TEST()
334 internal::ThreadWorkSource* tws; in TEST()
368 internal::ThreadWorkSource* tws; in TEST()
403 internal::ThreadWorkSource* tws; in TEST()
[all …]