Home
last modified time | relevance | path

Searched refs:SerialDeviceBatchScheduler (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dserial_device_batch_scheduler.h67 class SerialDeviceBatchScheduler : public std::enable_shared_from_this<
68 SerialDeviceBatchScheduler<TaskType>> {
70 ~SerialDeviceBatchScheduler();
104 std::shared_ptr<SerialDeviceBatchScheduler<TaskType>>* scheduler);
134 explicit SerialDeviceBatchScheduler(const Options& options);
191 TF_DISALLOW_COPY_AND_ASSIGN(SerialDeviceBatchScheduler);
204 typename SerialDeviceBatchScheduler<TaskType>::QueueOptions;
206 SDBSQueue(std::shared_ptr<SerialDeviceBatchScheduler<TaskType>> scheduler,
229 std::shared_ptr<SerialDeviceBatchScheduler<TaskType>> scheduler_;
262 Status SerialDeviceBatchScheduler<TaskType>::Create( in Create()
[all …]
Dserial_device_batch_scheduler_test.cc68 using Scheduler = SerialDeviceBatchScheduler<FakeTask>; in TEST()
93 SerialDeviceBatchScheduler<FakeTask>::Options options; in TEST()
118 std::shared_ptr<SerialDeviceBatchScheduler<FakeTask>> scheduler; in TEST()
120 SerialDeviceBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()
136 SerialDeviceBatchScheduler<FakeTask>::Options options; in TEST()
145 std::shared_ptr<SerialDeviceBatchScheduler<FakeTask>> scheduler; in TEST()
147 SerialDeviceBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()
202 SerialDeviceBatchScheduler<FakeTask>::Options options; in TEST()
229 std::shared_ptr<SerialDeviceBatchScheduler<FakeTask>> scheduler; in TEST()
231 SerialDeviceBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()
[all …]