Searched refs:combined_outputs (Results 1 – 1 of 1) sorted by relevance
407 Status SplitOutputTensors(const std::vector<Tensor>& combined_outputs, in SplitOutputTensors() argument429 DCHECK_EQ(batch->task(0).context->num_outputs(), combined_outputs.size()); in SplitOutputTensors()430 if (combined_outputs.size() != batch->task(0).context->num_outputs()) { in SplitOutputTensors()435 for (int i = 0; i < combined_outputs.size(); ++i) { in SplitOutputTensors()436 const Tensor& output_tensor = combined_outputs[i]; in SplitOutputTensors()519 std::vector<Tensor> combined_outputs; in ProcessFuncBatch() local531 opts, fhandle_, args, &combined_outputs, [&](const Status& run_status) { in ProcessFuncBatch()546 final_status = SplitOutputTensors(combined_outputs, batch.get()); in ProcessFuncBatch()