Home
last modified time | relevance | path

Searched refs:BatchTask (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc243 std::unique_ptr<BatchTask> batch_components(new BatchTask); in RegisterInput()
282 struct BatchTask : public serving::BatchTask { struct in tensorflow::BatchResource
294 using Batcher = serving::SharedBatchScheduler<BatchTask>;
295 using BatcherQueue = serving::BatchScheduler<BatchTask>;
296 using Batch = serving::Batch<BatchTask>;
302 const BatchTask& task = batch.task(task_idx); in ValidateBatch()
464 BatchTask& task = *(batch->mutable_task(j)); in SplitOutputTensors()
582 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
594 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
604 const BatchTask& task = batch->task(task_idx); in ProcessBatch()
[all …]
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler.h58 class BatchTask {
60 virtual ~BatchTask() = default;
Dbasic_batch_scheduler_test.cc30 class FakeTask : public BatchTask {
Dbatch_scheduler_test.cc26 class FakeTask : public BatchTask {
Dbasic_batch_scheduler_benchmark_test.cc84 class BenchmarkBatchTask : public BatchTask {
Dadaptive_shared_batch_scheduler_test.cc28 class FakeTask : public BatchTask {
Dserial_device_batch_scheduler_test.cc28 class FakeTask : public BatchTask {
Dshared_batch_scheduler_test.cc29 class FakeTask : public BatchTask {