Searched refs:task_sizes_plus_optional_padding (Results 1 – 1 of 1) sorted by relevance
415 std::vector<int64> task_sizes_plus_optional_padding; in SplitOutputTensors() local416 task_sizes_plus_optional_padding.reserve(batch->num_tasks()); in SplitOutputTensors()418 task_sizes_plus_optional_padding.push_back(batch->task(i).size()); in SplitOutputTensors()423 task_sizes_plus_optional_padding.push_back(padding_size); in SplitOutputTensors()449 output_tensor, task_sizes_plus_optional_padding, &split_tensor); in SplitOutputTensors()455 DCHECK_EQ(split_tensor.size(), task_sizes_plus_optional_padding.size()); in SplitOutputTensors()456 if (split_tensor.size() != task_sizes_plus_optional_padding.size()) { in SplitOutputTensors()460 task_sizes_plus_optional_padding.size()); in SplitOutputTensors()