Method
__init__
(self, dataset, *, layout=None, batch=False)
Source from the content-addressed store, hash-verified
| 903 | |
| 904 | class Input: |
| 905 | def __init__(self, dataset, *, layout=None, batch=False): |
| 906 | if not isinstance(dataset, dataset_ops.DatasetV2): |
| 907 | raise TypeError( |
| 908 | ( |
| 909 | "The inputs specified to DALIDataset must be instances of " |
| 910 | "type `tf.data.Dataset` got: `{}` of type: {} instead." |
| 911 | ).format(dataset, type(dataset)) |
| 912 | ) |
| 913 | self.dataset = dataset |
| 914 | self.layout = layout |
| 915 | self.batch = batch |
| 916 | |
| 917 | Input.__doc__ = _experimental_input_docstring |
| 918 | |
Tested by
no test coverage detected