Searched refs:BasicBatchScheduler (Results 1 – 3 of 3) sorted by relevance
138 class BasicBatchScheduler : public BatchScheduler<TaskType> {270 std::unique_ptr<BasicBatchScheduler>* scheduler);272 ~BasicBatchScheduler() override = default;283 explicit BasicBatchScheduler(290 TF_DISALLOW_COPY_AND_ASSIGN(BasicBatchScheduler);297 Status BasicBatchScheduler<TaskType>::Create( in Create()301 std::unique_ptr<BasicBatchScheduler>* scheduler) { in Create()336 new BasicBatchScheduler<TaskType>(std::move(shared_scheduler_queue))); in Create()341 Status BasicBatchScheduler<TaskType>::Schedule( in Schedule()347 size_t BasicBatchScheduler<TaskType>::NumEnqueuedTasks() const { in NumEnqueuedTasks()[all …]
111 const BasicBatchScheduler<BenchmarkBatchTask>::Options&129 const BasicBatchScheduler<BenchmarkBatchTask>::Options scheduler_options_;132 std::unique_ptr<BasicBatchScheduler<BenchmarkBatchTask>> scheduler_;136 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options) in ThroughputBenchmark()167 TF_CHECK_OK(BasicBatchScheduler<BenchmarkBatchTask>::Create( in ResetState()185 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options,206 const BasicBatchScheduler<BenchmarkBatchTask>::Options scheduler_options_;218 std::unique_ptr<BasicBatchScheduler<BenchmarkBatchTask>> scheduler_;231 const BasicBatchScheduler<BenchmarkBatchTask>::Options& scheduler_options, in LatencyBenchmark()296 TF_CHECK_OK(BasicBatchScheduler<BenchmarkBatchTask>::Create( in ResetState()[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()