Searched refs:FakeTask (Results 1 – 11 of 11) sorted by relevance
29 class FakeTask : public BatchTask { class31 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::__anonc47d7c6c0111::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 …]
29 class FakeTask : public BatchTask { class31 explicit FakeTask(size_t size) : size_(size) {} in FakeTask() function in tensorflow::serving::anonymous::FakeTask33 ~FakeTask() override = default;42 TF_DISALLOW_COPY_AND_ASSIGN(FakeTask);47 Status ScheduleTask(size_t task_size, BatchScheduler<FakeTask>* scheduler) { in ScheduleTask()48 std::unique_ptr<FakeTask> task(new FakeTask(task_size)); in ScheduleTask()71 using Scheduler = AdaptiveSharedBatchScheduler<FakeTask>; in TEST()101 AdaptiveSharedBatchScheduler<FakeTask>::Options options; in TEST()108 std::unique_ptr<Batch<FakeTask>> batch) { in TEST()124 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::__anonb1768b760111::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::__anon3e0f1e770111::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()
74 struct FakeTask { struct82 explicit FakeTaskRunner(std::vector<FakeTask>* tasks); argument93 std::vector<FakeTask>* tasks_;
70 FakeTaskRunner::FakeTaskRunner(std::vector<FakeTask>* tasks) : tasks_(tasks) {} in FakeTaskRunner()
91 std::vector<FakeTask> tasks; in main()
83 std::vector<FakeTask> tasks; in main()
13 class FakeTask : public TaskQueue::Task {15 FakeTask();
10 FakeTask::FakeTask() in FakeTask() function in base::sequence_manager::FakeTask