Searched refs:CreateDatasetContext (Results 1 – 2 of 2) sorted by relevance
318 Status DatasetOpsTestBase::CreateDatasetContext( in CreateDatasetContext() function in tensorflow::data::DatasetOpsTestBase835 TF_RETURN_IF_ERROR(CreateDatasetContext(dataset_kernel->get(), &inputs, in RunDatasetOp()940 TF_RETURN_IF_ERROR(CreateDatasetContext(dataset_kernel.get(), &inputs, in MakeDatasetTensor()
660 Status CreateDatasetContext(