Searched refs:SchedulingCapacity (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler_test.cc | 78 EXPECT_EQ(3 * 10, scheduler->SchedulingCapacity()); in TEST()
81 EXPECT_EQ((3 * 10) - 3, scheduler->SchedulingCapacity()); in TEST()
84 EXPECT_EQ((3 * 10) - (3 + 5), scheduler->SchedulingCapacity()); in TEST()
|
| D | basic_batch_scheduler.h | 276 size_t SchedulingCapacity() const override;
352 size_t BasicBatchScheduler<TaskType>::SchedulingCapacity() const { in SchedulingCapacity() function
353 return shared_scheduler_queue_->SchedulingCapacity(); in SchedulingCapacity()
|
| D | shared_batch_scheduler.h | 297 size_t SchedulingCapacity() const;
408 size_t SchedulingCapacity() const override;
776 size_t Queue<TaskType>::SchedulingCapacity() const { in SchedulingCapacity() function
911 size_t QueueHandle<TaskType>::SchedulingCapacity() const { in SchedulingCapacity() function
912 return queue_->SchedulingCapacity(); in SchedulingCapacity()
|
| D | shared_batch_scheduler_test.cc | 433 EXPECT_EQ(max_enqueued_batches * 2, queue->SchedulingCapacity()); in TEST()
445 EXPECT_EQ((max_enqueued_batches - i) * 2, queue->SchedulingCapacity()); in TEST()
449 queue->SchedulingCapacity()); in TEST()
453 EXPECT_EQ(0, queue->SchedulingCapacity()); in TEST()
460 EXPECT_EQ(0, queue->SchedulingCapacity()); in TEST()
|
| D | serial_device_batch_scheduler_test.cc | 380 EXPECT_EQ(queue2->SchedulingCapacity(), 9 * 1000 + 900); in TEST()
385 EXPECT_EQ(queue2->SchedulingCapacity(), 9 * 1000 + 600); in TEST()
389 EXPECT_EQ(queue2->SchedulingCapacity(), 8 * 1000 + 300); in TEST()
|
| D | adaptive_shared_batch_scheduler_test.cc | 345 EXPECT_EQ(queue->SchedulingCapacity(), 9 * 1000 + 900); in TEST()
350 EXPECT_EQ(queue->SchedulingCapacity(), 9 * 1000 + 600); in TEST()
355 EXPECT_EQ(queue->SchedulingCapacity(), 9 * 1000 + 300); in TEST()
|
| D | batch_scheduler.h | 191 virtual size_t SchedulingCapacity() const = 0;
|
| D | serial_device_batch_scheduler.h | 220 size_t SchedulingCapacity() const override;
536 size_t SDBSQueue<TaskType>::SchedulingCapacity() const { in SchedulingCapacity() function
|
| D | adaptive_shared_batch_scheduler.h | 278 size_t SchedulingCapacity() const override;
730 size_t ASBSQueue<TaskType>::SchedulingCapacity() const { in SchedulingCapacity() function
|