Home
last modified time | relevance | path

Searched refs:CreateOpKernelContext (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc332 TF_RETURN_IF_ERROR(CreateOpKernelContext( in CreateDatasetContext()
488 Status DatasetOpsTestBase::CreateOpKernelContext( in CreateOpKernelContext() function in tensorflow::data::DatasetOpsTestBase
491 return CreateOpKernelContext(kernel, inputs, &params_, context); in CreateOpKernelContext()
494 Status DatasetOpsTestBase::CreateOpKernelContext( in CreateOpKernelContext() function in tensorflow::data::DatasetOpsTestBase
H A Ddataset_test_base.h746 Status CreateOpKernelContext(OpKernel* kernel,
751 Status CreateOpKernelContext(OpKernel* kernel,
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_defun_op_test.cc111 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateMapDefunContext()