Searched defs:input_task_size (Results 1 – 4 of 4) sorted by relevance
27 int compute_task_size_from_open_batch(int 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()43 int compute_tail_task_size(int task_size_from_open_batch, int input_task_size, in compute_tail_task_size()63 InputSplitMetadata::InputSplitMetadata(int input_task_size, in InputSplitMetadata()78 int input_task_size, int open_batch_remaining_slot, in generate_task_sizes()
905 const int64_t input_task_size = (*task)->size(); in ScheduleWithoutOrEagerSplit() local
158 const int input_task_size = owned_input_task->size(); in get_split_func() local
493 const int64_t input_task_size = input_task.size(); in SplitInputTask() local