Home
last modified time | relevance | path

Searched refs:BasicBatchScheduler (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbasic_batch_scheduler.h138 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 …]
H A Dbasic_batch_scheduler_benchmark_test.cc113 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 …]
H A Dbasic_batch_scheduler_test.cc68 BasicBatchScheduler<FakeTask>::Options options; in TEST()
73 std::unique_ptr<BasicBatchScheduler<FakeTask>> scheduler; in TEST()
75 BasicBatchScheduler<FakeTask>::Create(options, callback, &scheduler)); in TEST()