Searched refs:GetDatasetFromContext (Results 1 – 2 of 2) sorted by relevance
85 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
107 GetDatasetFromContext(range_context.get(), 0, range_dataset)); in CreateRangeDataset()123 Status GetDatasetFromContext(OpKernelContext* context, int output_index,