Searched refs:_RestructuredDataset (Results 1 – 2 of 2) sorted by relevance
554 dataset = dataset_ops._RestructuredDataset(dataset, new_structure)
4786 return _RestructuredDataset(normalized_dataset, dataset.element_spec)4789 class _RestructuredDataset(UnaryDataset): class4797 super(_RestructuredDataset, self).__init__(dataset, variant_tensor)