Searched refs:last_task_context (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 561 OpKernelContext* last_task_context = last_task.context; in ProcessFuncBatch() local
593 status = ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessFuncBatch()
606 const string& model_name = GetModelName(last_task_context); in ProcessFuncBatch()
610 last_task_context->op_kernel().name_view().data()); in ProcessFuncBatch()
643 OpKernelContext* last_task_context = in ProcessBatch() local
648 OP_REQUIRES_OK_ASYNC(last_task_context, ValidateBatch(*batch), in ProcessBatch()
655 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessBatch()
656 OP_REQUIRES_OK_ASYNC(last_task_context, concat_status, last_task_callback); in ProcessBatch()
660 last_task_context->set_output(i, concatenated_tensors[i]); in ProcessBatch()
694 last_task_context, in ProcessBatch()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 157 auto* last_task_context = last_task.context; in ProcessFuncBatchImpl() local
159 opts.step_container = last_task_context->step_container(); in ProcessFuncBatchImpl()
160 opts.cancellation_manager = last_task_context->cancellation_manager(); in ProcessFuncBatchImpl()
161 opts.collective_executor = last_task_context->collective_executor(); in ProcessFuncBatchImpl()
162 opts.stats_collector = last_task_context->stats_collector(); in ProcessFuncBatchImpl()
163 opts.runner = last_task_context->runner(); in ProcessFuncBatchImpl()
164 opts.run_all_kernels_inline = last_task_context->run_all_kernels_inline(); in ProcessFuncBatchImpl()
|