Searched refs:task_sizes_plus_optional_padding (Results 1 – 1 of 1) sorted by relevance
592 std::vector<int64_t> task_sizes_plus_optional_padding; in SplitOutputTensors() local593 task_sizes_plus_optional_padding.reserve(batch->num_tasks()); in SplitOutputTensors()595 task_sizes_plus_optional_padding.push_back(batch->task(i).size()); in SplitOutputTensors()600 task_sizes_plus_optional_padding.push_back(padding_size); in SplitOutputTensors()628 output_tensor, task_sizes_plus_optional_padding, &split_tensor); in SplitOutputTensors()634 DCHECK_EQ(split_tensor.size(), task_sizes_plus_optional_padding.size()); in SplitOutputTensors()635 if (split_tensor.size() != task_sizes_plus_optional_padding.size()) { in SplitOutputTensors()639 task_sizes_plus_optional_padding.size()); in SplitOutputTensors()