Searched refs:concatenated_tensors (Results 1 – 1 of 1) sorted by relevance
331 std::vector<Tensor>* concatenated_tensors) const { in ConcatInputTensors()341 concatenated_tensors->reserve(num_inputs); in ConcatInputTensors()402 concatenated_tensors->push_back(concatenated_tensor); in ConcatInputTensors()504 std::vector<Tensor> concatenated_tensors; in ProcessFuncBatch() local506 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessFuncBatch()521 std::vector<Tensor> args(concatenated_tensors.begin(), in ProcessFuncBatch()522 concatenated_tensors.end()); in ProcessFuncBatch()571 std::vector<Tensor> concatenated_tensors; in ProcessBatch() local573 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessBatch()578 last_task_context->set_output(i, concatenated_tensors.at(i)); in ProcessBatch()