| 374 | } |
| 375 | |
| 376 | Status DatasetOpsTestBase::CreateIteratorContext( |
| 377 | OpKernelContext* const op_context, |
| 378 | std::unique_ptr<IteratorContext>* iterator_context) { |
| 379 | IteratorContext::Params params(op_context); |
| 380 | params.resource_mgr = op_context->resource_manager(); |
| 381 | function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); |
| 382 | params.function_handle_cache = function_handle_cache_.get(); |
| 383 | params.cancellation_manager = cancellation_manager_.get(); |
| 384 | *iterator_context = absl::make_unique<IteratorContext>(params); |
| 385 | return Status::OK(); |
| 386 | } |
| 387 | |
| 388 | Status DatasetOpsTestBase::GetDatasetFromContext(OpKernelContext* context, |
| 389 | int output_index, |
nothing calls this directly
no test coverage detected