Home
last modified time | relevance | path

Searched refs:concatenated_tensors (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.cc423 std::vector<Tensor>* concatenated_tensors) const { in ConcatInputTensors()
448 concatenated_tensors->reserve(num_inputs); in ConcatInputTensors()
484 concatenated_tensors->push_back(concatenated_tensor); in ConcatInputTensors()
711 std::vector<Tensor> concatenated_tensors; in ProcessFuncBatch() local
712 status = ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessFuncBatch()
719 std::vector<Tensor> args(concatenated_tensors.begin(), in ProcessFuncBatch()
720 concatenated_tensors.end()); in ProcessFuncBatch()
790 std::vector<Tensor> concatenated_tensors; in ProcessBatch() local
792 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessBatch()
798 last_task_context->set_output(i, concatenated_tensors[i]); in ProcessBatch()
H A Dbatch_resource_base.h221 std::vector<Tensor>* concatenated_tensors) const;