Searched refs:ProcessBatch (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler_benchmark_test.cc | 126 void ProcessBatch(std::unique_ptr<Batch<BenchmarkBatchTask>> batch);
165 ProcessBatch(std::move(batch)); in ResetState()
171 void ThroughputBenchmark::ProcessBatch( in ProcessBatch() function in tensorflow::serving::__anon9dfe3e810111::ThroughputBenchmark
200 void ProcessBatch(std::unique_ptr<Batch<BenchmarkBatchTask>> batch);
294 ProcessBatch(std::move(batch)); in ResetState()
306 void LatencyBenchmark::ProcessBatch( in ProcessBatch() function in tensorflow::serving::__anon9dfe3e810111::LatencyBenchmark
|
| D | shared_batch_scheduler.h | 320 void ProcessBatch(std::unique_ptr<Batch<TaskType>> batch);
584 queue_for_batch->ProcessBatch(std::move(batch_to_process)); in ThreadLogic()
814 void Queue<TaskType>::ProcessBatch(std::unique_ptr<Batch<TaskType>> batch) { in ProcessBatch() function
|
| D | batch_resource_base.h | 196 void ProcessBatch(std::unique_ptr<BatchT> batch) const;
|
| D | batch_resource_base.cc | 636 void BatchResourceBase::ProcessBatch(std::unique_ptr<BatchT> batch) const { in ProcessBatch() function in tensorflow::serving::BatchResourceBase
739 ProcessBatch(std::move(batch)); in LookupOrCreateBatcherQueue()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.h | 323 Status ProcessBatch(int64 batch_size, int64 num_elements, bool drop_remainder,
|
| D | parallel_batch_dataset_op.cc | 225 ProcessBatch(dataset()->batch_size_, result->num_elements, in GetNextInternal()
|
| D | dataset_utils.cc | 804 Status ProcessBatch(int64 batch_size, int64 num_elements, bool drop_remainder, in ProcessBatch() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 263 ProcessBatch(dataset()->batch_size_, result->num_elements, in GetNextInternal()
|