Searched refs:concatenated_tensors (Results 1 – 2 of 2) sorted by relevance
319 std::vector<Tensor>* concatenated_tensors) const { in ConcatInputTensors()338 concatenated_tensors->reserve(num_inputs); in ConcatInputTensors()374 concatenated_tensors->push_back(concatenated_tensor); in ConcatInputTensors()592 std::vector<Tensor> concatenated_tensors; in ProcessFuncBatch() local593 status = ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessFuncBatch()599 std::vector<Tensor> args(concatenated_tensors.begin(), in ProcessFuncBatch()600 concatenated_tensors.end()); in ProcessFuncBatch()653 std::vector<Tensor> concatenated_tensors; in ProcessBatch() local655 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessBatch()660 last_task_context->set_output(i, concatenated_tensors[i]); in ProcessBatch()
168 std::vector<Tensor>* concatenated_tensors) const;