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 | basic_batch_scheduler.h | 192 size_t NumEnqueuedTasks() const override;
251 size_t BasicBatchScheduler<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
252 return shared_scheduler_queue_->NumEnqueuedTasks(); in NumEnqueuedTasks()
|
| D | shared_batch_scheduler.h | 246 size_t NumEnqueuedTasks() const;
342 size_t NumEnqueuedTasks() const override;
570 size_t Queue<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
692 size_t QueueHandle<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
693 return 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 | adaptive_shared_batch_scheduler_test.cc | 371 EXPECT_EQ(queue->NumEnqueuedTasks(), 1); in TEST()
376 EXPECT_EQ(queue->NumEnqueuedTasks(), 3); in TEST()
380 EXPECT_EQ(queue->NumEnqueuedTasks(), 4); in TEST()
|
| 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 | 166 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 | 248 size_t NumEnqueuedTasks() const override;
595 size_t ASBSQueue<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks() function
|