Searched refs:_ModelDataset (Results 1 – 2 of 2) sorted by relevance
63 return dataset_ops._ModelDataset(dataset) # pylint: disable=protected-access
205 dataset = _ModelDataset(dataset, cpu_budget)3283 class _ModelDataset(UnaryUnchangedStructureDataset): class3292 super(_ModelDataset, self).__init__(input_dataset, variant_tensor)