Searched refs:SharedBatchScheduler (Results 1 – 4 of 4) sorted by relevance
92 SharedBatchScheduler<FakeTask>::Options options; in TEST()94 std::shared_ptr<SharedBatchScheduler<FakeTask>> scheduler; in TEST()96 SharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()99 SharedBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST()153 SharedBatchScheduler<FakeTask>::Options options; in TEST()155 std::shared_ptr<SharedBatchScheduler<FakeTask>> scheduler; in TEST()156 TF_ASSERT_OK(SharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()157 SharedBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST()214 SharedBatchScheduler<FakeTask>::Options options; in TEST()217 std::shared_ptr<SharedBatchScheduler<FakeTask>> scheduler; in TEST()[all …]
108 class SharedBatchScheduler109 : public std::enable_shared_from_this<SharedBatchScheduler<TaskType>> {128 std::shared_ptr<SharedBatchScheduler<TaskType>>* scheduler);130 ~SharedBatchScheduler();205 explicit SharedBatchScheduler(const Options& options);238 TF_DISALLOW_COPY_AND_ASSIGN(SharedBatchScheduler);274 Queue(const typename SharedBatchScheduler<TaskType>::QueueOptions& options,349 const typename SharedBatchScheduler<TaskType>::QueueOptions options_;402 QueueHandle(std::shared_ptr<SharedBatchScheduler<TaskType>> scheduler,414 std::shared_ptr<SharedBatchScheduler<TaskType>> scheduler_;[all …]
173 std::shared_ptr<SharedBatchScheduler<TaskType>> shared_batch_scheduler =302 std::shared_ptr<SharedBatchScheduler<TaskType>> shared_scheduler; in Create()305 typename SharedBatchScheduler<TaskType>::Options shared_scheduler_options; in Create()310 TF_RETURN_IF_ERROR(SharedBatchScheduler<TaskType>::Create( in Create()316 typename SharedBatchScheduler<TaskType>::QueueOptions in Create()
110 using BatcherT = SharedBatchScheduler<BatchResourceBase::BatchTask>;