Searched refs:_ModelDataset (Results 1 – 2 of 2) sorted by relevance
34 return dataset_ops._ModelDataset(dataset) # pylint: disable=protected-access
380 dataset = _ModelDataset(dataset, algorithm, cpu_budget, ram_budget)4644 class _ModelDataset(UnaryUnchangedStructureDataset): class4655 super(_ModelDataset, self).__init__(input_dataset, variant_tensor)