Searched refs:FakeTask (Results 1 – 7 of 7) sorted by relevance
29 class FakeTask : public BatchTask { class31 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::__anon758eaa850111::FakeTask33 ~FakeTask() override = default;40 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);45 Status ScheduleTask(size_t task_size, BatchScheduler<FakeTask>* scheduler) { in ScheduleTask()46 std::unique_ptr<FakeTask> task(new FakeTask(task_size)); in ScheduleTask()74 [&queue_0_callback_called](std::unique_ptr<Batch<FakeTask>> batch) { in TEST()84 [&queue_1_callback_called](std::unique_ptr<Batch<FakeTask>> batch) { in TEST()92 SharedBatchScheduler<FakeTask>::Options options; in TEST()94 std::shared_ptr<SharedBatchScheduler<FakeTask>> scheduler; in TEST()[all …]
28 class FakeTask : public BatchTask { class30 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::anonymous::FakeTask32 ~FakeTask() override = default;39 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);44 Status ScheduleTask(size_t task_size, BatchScheduler<FakeTask>* scheduler) { in ScheduleTask()45 std::unique_ptr<FakeTask> task(new FakeTask(task_size)); in ScheduleTask()68 using Scheduler = SerialDeviceBatchScheduler<FakeTask>; in TEST()93 SerialDeviceBatchScheduler<FakeTask>::Options options; in TEST()102 std::unique_ptr<Batch<FakeTask>> batch) { in TEST()118 std::shared_ptr<SerialDeviceBatchScheduler<FakeTask>> scheduler; in TEST()[all …]
28 class FakeTask : public BatchTask { class30 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::anonymous::FakeTask32 ~FakeTask() override = default;39 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);44 Status ScheduleTask(size_t task_size, BatchScheduler<FakeTask>* scheduler) { in ScheduleTask()45 std::unique_ptr<FakeTask> task(new FakeTask(task_size)); in ScheduleTask()68 using Scheduler = AdaptiveSharedBatchScheduler<FakeTask>; in TEST()98 AdaptiveSharedBatchScheduler<FakeTask>::Options options; in TEST()105 std::unique_ptr<Batch<FakeTask>> batch) { in TEST()121 std::shared_ptr<AdaptiveSharedBatchScheduler<FakeTask>> scheduler; in TEST()[all …]
26 class FakeTask : public BatchTask { class28 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::__anon33b053af0111::FakeTask30 ~FakeTask() override = default;37 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);41 Batch<FakeTask> batch; in TEST()48 auto task0 = new FakeTask(3); in TEST()49 batch.AddTask(std::unique_ptr<FakeTask>(task0)); in TEST()57 auto task1 = new FakeTask(7); in TEST()58 batch.AddTask(std::unique_ptr<FakeTask>(task1)); in TEST()84 Batch<FakeTask> batch; in TEST()[all …]
30 class FakeTask : public BatchTask { class32 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::__anon6a366d700111::FakeTask34 ~FakeTask() override = default;41 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);46 Status ScheduleTask(size_t task_size, BatchScheduler<FakeTask>* scheduler) { in ScheduleTask()47 std::unique_ptr<FakeTask> task(new FakeTask(task_size)); in ScheduleTask()60 auto callback = [&callback_called](std::unique_ptr<Batch<FakeTask>> batch) { in TEST()68 BasicBatchScheduler<FakeTask>::Options options; in TEST()73 std::unique_ptr<BasicBatchScheduler<FakeTask>> scheduler; in TEST()75 BasicBatchScheduler<FakeTask>::Create(options, callback, &scheduler)); in TEST()
13 class FakeTask : public TaskQueue::Task {15 FakeTask();
10 FakeTask::FakeTask() in FakeTask() function in base::sequence_manager::FakeTask