Searched refs:last_task_context (Results 1 – 1 of 1) sorted by relevance
478 OpKernelContext* last_task_context = in ProcessFuncBatch() local506 ConcatInputTensors(*batch, last_task_context, &concatenated_tensors); in ProcessFuncBatch()511 opts.step_id = last_task_context->step_id(); in ProcessFuncBatch()512 opts.step_container = last_task_context->step_container(); in ProcessFuncBatch()513 opts.cancellation_manager = last_task_context->cancellation_manager(); in ProcessFuncBatch()514 opts.stats_collector = last_task_context->stats_collector(); in ProcessFuncBatch()515 opts.rendezvous = last_task_context->rendezvous(); in ProcessFuncBatch()516 opts.runner = last_task_context->runner(); in ProcessFuncBatch()518 auto* flib = last_task_context->function_library(); in ProcessFuncBatch()561 OpKernelContext* last_task_context = in ProcessBatch() local[all …]