A `Dataset` that acts as an identity, and applies optimizations.
| 3842 | |
| 3843 | |
| 3844 | class _OptimizeDataset(UnaryUnchangedStructureDataset): |
| 3845 | """A `Dataset` that acts as an identity, and applies optimizations.""" |
| 3846 | |
| 3847 | def __init__(self, input_dataset, optimizations, optimization_configs=None): |
| 3848 | self._input_dataset = input_dataset |
| 3849 | if optimizations is None: |
| 3850 | optimizations = [] |
| 3851 | if optimization_configs is None: |
| 3852 | optimization_configs = [] |
| 3853 | self._optimizations = ops.convert_to_tensor( |
| 3854 | optimizations, dtype=dtypes.string, name="optimizations") |
| 3855 | variant_tensor = gen_dataset_ops.optimize_dataset( |
| 3856 | input_dataset._variant_tensor, # pylint: disable=protected-access |
| 3857 | self._optimizations, |
| 3858 | optimization_configs=optimization_configs, |
| 3859 | **self._flat_structure) |
| 3860 | super(_OptimizeDataset, self).__init__(input_dataset, variant_tensor) |
| 3861 | |
| 3862 | |
| 3863 | class _SetStatsAggregatorDataset(UnaryUnchangedStructureDataset): |