Searched refs:BasicBatchScheduler (Results 1 – 3 of 3) sorted by relevance
138 class BasicBatchScheduler : public BatchScheduler<TaskType> {276 std::unique_ptr<BasicBatchScheduler>* scheduler);278 ~BasicBatchScheduler() override = default;289 explicit BasicBatchScheduler(296 TF_DISALLOW_COPY_AND_ASSIGN(BasicBatchScheduler);303 Status BasicBatchScheduler<TaskType>::Create( in Create()307 std::unique_ptr<BasicBatchScheduler>* scheduler) { in Create()343 new BasicBatchScheduler<TaskType>(std::move(shared_scheduler_queue))); in Create()348 Status BasicBatchScheduler<TaskType>::Schedule( in Schedule()354 size_t BasicBatchScheduler<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks()[all …]
113 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options) in ThroughputBenchmark()119 TF_CHECK_OK(BasicBatchScheduler<BenchmarkBatchTask>::Create( in ThroughputBenchmark()126 BasicBatchScheduler<BenchmarkBatchTask>* GetScheduler() const { in GetScheduler()142 const BasicBatchScheduler<BenchmarkBatchTask>::Options scheduler_options_;145 std::unique_ptr<BasicBatchScheduler<BenchmarkBatchTask>> scheduler_;154 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options,179 const BasicBatchScheduler<BenchmarkBatchTask>::Options scheduler_options_;191 std::unique_ptr<BasicBatchScheduler<BenchmarkBatchTask>> scheduler_;204 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options, in LatencyBenchmark()213 TF_CHECK_OK(BasicBatchScheduler<BenchmarkBatchTask>::Create( in LatencyBenchmark()[all …]
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()