Searched refs:output_task_sizes (Results 1 – 1 of 1) sorted by relevance
420 std::vector<int64> output_task_sizes; in SplitInputTask() local423 output_task_sizes.push_back(open_batch_remaining_slot); in SplitInputTask()429 output_task_sizes.push_back(next_task_size); in SplitInputTask()432 const int output_task_num = output_task_sizes.size(); in SplitInputTask()455 output_task_sizes, &split_tensors); in SplitInputTask()461 if (split_tensors.size() != output_task_sizes.size()) { in SplitInputTask()465 split_tensors.size(), " splits; expected ", output_task_sizes.size()); in SplitInputTask()