Searched refs:AdaptiveSharedBatchScheduler (Results 1 – 4 of 4) sorted by relevance
71 using Scheduler = AdaptiveSharedBatchScheduler<FakeTask>; in TEST()101 AdaptiveSharedBatchScheduler<FakeTask>::Options options; in TEST()124 std::shared_ptr<AdaptiveSharedBatchScheduler<FakeTask>> scheduler; in TEST()126 AdaptiveSharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()146 AdaptiveSharedBatchScheduler<FakeTask>::Options options; in TEST()167 std::shared_ptr<AdaptiveSharedBatchScheduler<FakeTask>> scheduler; in TEST()169 AdaptiveSharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()208 AdaptiveSharedBatchScheduler<FakeTask>::Options options; in TEST()240 std::shared_ptr<AdaptiveSharedBatchScheduler<FakeTask>> scheduler; in TEST()242 AdaptiveSharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST()[all …]
79 class AdaptiveSharedBatchScheduler81 AdaptiveSharedBatchScheduler<TaskType>> {83 ~AdaptiveSharedBatchScheduler() { in ~AdaptiveSharedBatchScheduler()130 std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>>* scheduler);176 explicit AdaptiveSharedBatchScheduler(const Options& options);249 TF_DISALLOW_COPY_AND_ASSIGN(AdaptiveSharedBatchScheduler);262 typename AdaptiveSharedBatchScheduler<TaskType>::QueueOptions;264 ASBSQueue(std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>> scheduler,294 std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>> scheduler_;337 constexpr double AdaptiveSharedBatchScheduler<TaskType>::kMaxStepSizeMultiplier;[all …]
112 AdaptiveSharedBatchScheduler<BatchResourceBase::BatchTask>;
262 serving::AdaptiveSharedBatchScheduler< in ComputeAsync()