Searched refs:output_task_sizes (Results 1 – 1 of 1) sorted by relevance
535 std::vector<int64_t> output_task_sizes; in SplitInputTask() local536 output_task_sizes.resize(num_batches); in SplitInputTask()538 output_task_sizes[i] = task_sizes[i]; in SplitInputTask()562 output_task_sizes, &split_tensors); in SplitInputTask()568 if (split_tensors.size() != output_task_sizes.size()) { in SplitInputTask()572 split_tensors.size(), " splits; expected ", output_task_sizes.size()); in SplitInputTask()