Home
last modified time | relevance | path

Searched refs:input_task_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc383 const int64 input_task_size = input_task.size(); in SplitInputTask() local
385 DCHECK_GT(input_task_size, open_batch_remaining_slot); in SplitInputTask()
426 for (int left_task_size = input_task_size - open_batch_remaining_slot; in SplitInputTask()
Dshared_batch_scheduler.h721 const int64 input_task_size = (*task)->size(); in ScheduleWithSplit() local
725 if (input_task_size <= open_batch_remaining_slot) { in ScheduleWithSplit()