Searched refs:task_sizes_plus_optional_padding (Results 1 – 1 of 1) sorted by relevance
485 std::vector<int64> task_sizes_plus_optional_padding; in SplitOutputTensors() local486 task_sizes_plus_optional_padding.reserve(batch->num_tasks()); in SplitOutputTensors()488 task_sizes_plus_optional_padding.push_back(batch->task(i).size()); in SplitOutputTensors()493 task_sizes_plus_optional_padding.push_back(padding_size); in SplitOutputTensors()521 output_tensor, task_sizes_plus_optional_padding, &split_tensor); in SplitOutputTensors()527 DCHECK_EQ(split_tensor.size(), task_sizes_plus_optional_padding.size()); in SplitOutputTensors()528 if (split_tensor.size() != task_sizes_plus_optional_padding.size()) { in SplitOutputTensors()532 task_sizes_plus_optional_padding.size()); in SplitOutputTensors()