Searched refs:open_batch_remaining_slot (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 240 int open_batch_remaining_slot, int max_batch_size, in GetBatcherQueueOptions()
242 return SplitInputTask(input_task, open_batch_remaining_slot, in GetBatcherQueueOptions()
276 int open_batch_remaining_slot, int max_batch_size, in GetAdaptiveBatcherQueueOptions()
278 return SplitInputTask(input_task, open_batch_remaining_slot, in GetAdaptiveBatcherQueueOptions()
380 std::unique_ptr<BatchTask>* input_task_ptr, int open_batch_remaining_slot, in SplitInputTask() argument
385 DCHECK_GT(input_task_size, open_batch_remaining_slot); in SplitInputTask()
422 if (open_batch_remaining_slot > 0) { in SplitInputTask()
423 output_task_sizes.push_back(open_batch_remaining_slot); in SplitInputTask()
426 for (int left_task_size = input_task_size - open_batch_remaining_slot; in SplitInputTask()
|
| D | shared_batch_scheduler.h | 271 std::unique_ptr<TaskType>* input_task, int open_batch_remaining_slot,
718 const int64 open_batch_remaining_slot = in ScheduleWithSplit() local
725 if (input_task_size <= open_batch_remaining_slot) { in ScheduleWithSplit()
871 const int open_batch_remaining_slot = in SplitInputBatchIntoSubtasks() local
874 std::move(input_task), open_batch_remaining_slot, in SplitInputBatchIntoSubtasks()
|
| D | batch_resource_base.h | 186 std::unique_ptr<BatchTask>* input_task_ptr, int open_batch_remaining_slot,
|