Home
last modified time | relevance | path

Searched refs:GetDatasetFromContext (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc85 GetDatasetFromContext(context.get(), 0, tensor_slice_dataset)); in CreateTensorSliceDataset()
105 TF_RETURN_IF_ERROR(GetDatasetFromContext(context, 0, dataset)); in CreateDataset()
119 Status DatasetOpsTestBase::GetDatasetFromContext(OpKernelContext* context, in GetDatasetFromContext() function in tensorflow::data::DatasetOpsTestBase
Ddataset_test_base.h107 GetDatasetFromContext(range_context.get(), 0, range_dataset)); in CreateRangeDataset()
123 Status GetDatasetFromContext(OpKernelContext* context, int output_index,