Searched refs:input_task_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | input_split_metadata.cc | 27 int compute_task_size_from_open_batch(int input_task_size, in compute_task_size_from_open_batch() argument
31 ? (input_task_size + batch_size_limit - open_batch_remaining_slot) in compute_task_size_from_open_batch()
32 : input_task_size; in compute_task_size_from_open_batch()
35 int compute_head_task_size(int input_task_size, int open_batch_remaining_slot, in compute_head_task_size() argument
38 return std::min(input_task_size, batch_size_limit); in compute_head_task_size()
40 return std::min(open_batch_remaining_slot, input_task_size); in compute_head_task_size()
43 int compute_tail_task_size(int task_size_from_open_batch, int input_task_size, in compute_tail_task_size() argument
47 if (input_task_size <= open_batch_remaining_slot) { in compute_tail_task_size()
48 tail_task_size = input_task_size; in compute_tail_task_size()
63 InputSplitMetadata::InputSplitMetadata(int input_task_size, in InputSplitMetadata() argument
[all …]
|
| D | input_split_metadata.h | 35 InputSplitMetadata(int input_task_size, int open_batch_remaining_slot,
45 absl::FixedArray<int> generate_task_sizes(int input_task_size,
|
| D | batch_resource_base.cc | 493 const int64_t input_task_size = input_task.size(); in SplitInputTask() local
495 DCHECK_GT(input_task_size, 0); in SplitInputTask()
531 input_task_size, open_batch_remaining_slot, max_batch_size); in SplitInputTask()
|
| D | shared_batch_scheduler_test.cc | 158 const int input_task_size = owned_input_task->size(); in get_split_func() local
161 input_task_size, open_batch_remaining_slot, max_batch_size); in get_split_func()
|
| D | shared_batch_scheduler.h | 905 const int64_t input_task_size = (*task)->size(); in ScheduleWithoutOrEagerSplit() local
909 if (input_task_size <= open_batch_remaining_slot || in ScheduleWithoutOrEagerSplit()
|