Searched refs:combined_outputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 478 const std::vector<Tensor>& combined_outputs, BatchT* batch) const { in SplitOutputTensors() argument
499 DCHECK_EQ(batch->task(0).context->num_outputs(), combined_outputs.size()); in SplitOutputTensors()
500 int combined_outputs_size = combined_outputs.size(); in SplitOutputTensors()
506 for (int i = 0, iter_limit = combined_outputs.size(); i < iter_limit; ++i) { in SplitOutputTensors()
507 const Tensor& output_tensor = combined_outputs[i]; in SplitOutputTensors()
598 std::vector<Tensor> combined_outputs; in ProcessFuncBatch() local
616 last_task, args, &combined_outputs, [&](const Status& run_status) { in ProcessFuncBatch()
631 final_status = SplitOutputTensors(combined_outputs, batch.get()); in ProcessFuncBatch()
|
| D | batch_resource_base.h | 150 absl::Span<const Tensor> inputs, std::vector<Tensor>* combined_outputs,
190 Status SplitOutputTensors(const std::vector<Tensor>& combined_outputs,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 155 std::vector<Tensor>* combined_outputs, in ProcessFuncBatchImpl() argument
171 flib_->Run(opts, fhandle_, inputs, combined_outputs, in ProcessFuncBatchImpl()
|