Searched refs:NumEnqueuedTasks (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler_test.cc | 77 EXPECT_EQ(0, scheduler->NumEnqueuedTasks()); in TEST()
80 EXPECT_EQ(1, scheduler->NumEnqueuedTasks()); in TEST()
83 EXPECT_EQ(2, scheduler->NumEnqueuedTasks()); in TEST()
|
| D | adaptive_shared_batch_scheduler_test.cc | 132 while (queue->NumEnqueuedTasks() > 0) { in TEST()
135 while (queue->NumEnqueuedTasks() > 0) { in TEST()
253 while (queue1->NumEnqueuedTasks() > 0) { in TEST()
306 while (queue->NumEnqueuedTasks() > 0) { in TEST()
340 while (queue->NumEnqueuedTasks() > 0) { in TEST()
344 EXPECT_EQ(queue->NumEnqueuedTasks(), 1); in TEST()
349 EXPECT_EQ(queue->NumEnqueuedTasks(), 3); in TEST()
354 EXPECT_EQ(queue->NumEnqueuedTasks(), 1); in TEST()
|
| D | basic_batch_scheduler.h | 275 size_t NumEnqueuedTasks() const override;
347 size_t BasicBatchScheduler<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
348 return shared_scheduler_queue_->NumEnqueuedTasks(); in NumEnqueuedTasks()
|
| D | shared_batch_scheduler_test.cc | 432 EXPECT_EQ(0, queue->NumEnqueuedTasks()); in TEST()
439 EXPECT_EQ(0, queue->NumEnqueuedTasks()); in TEST()
444 EXPECT_EQ(i * 2, queue->NumEnqueuedTasks()); in TEST()
447 EXPECT_EQ((i * 2) + 1, queue->NumEnqueuedTasks()); in TEST()
452 EXPECT_EQ(max_enqueued_batches * 2, queue->NumEnqueuedTasks()); in TEST()
459 EXPECT_EQ(max_enqueued_batches * 2, queue->NumEnqueuedTasks()); in TEST()
|
| D | shared_batch_scheduler.h | 293 size_t NumEnqueuedTasks() const;
407 size_t NumEnqueuedTasks() const override;
766 size_t Queue<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
906 size_t QueueHandle<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
907 return queue_->NumEnqueuedTasks(); in NumEnqueuedTasks()
|
| D | serial_device_batch_scheduler_test.cc | 379 EXPECT_EQ(queue2->NumEnqueuedTasks(), 1); in TEST()
384 EXPECT_EQ(queue2->NumEnqueuedTasks(), 3); in TEST()
388 EXPECT_EQ(queue2->NumEnqueuedTasks(), 4); in TEST()
|
| D | batch_scheduler.h | 177 virtual size_t NumEnqueuedTasks() const = 0;
|
| D | serial_device_batch_scheduler.h | 217 size_t NumEnqueuedTasks() const override;
530 size_t SDBSQueue<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
|
| D | adaptive_shared_batch_scheduler.h | 275 size_t NumEnqueuedTasks() const override;
724 size_t ASBSQueue<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
|