Searched refs:_UnbatchDataset (Results 1 – 1 of 1) sorted by relevance
2243 return _UnbatchDataset(normalized_dataset)4804 class _UnbatchDataset(UnaryDataset): class4826 super(_UnbatchDataset, self).__init__(input_dataset, variant_tensor)