Returns a tf.dataset.Dataset.
(self, input_context: Optional[tf.distribute.InputContext] = None)
| 120 | return ret |
| 121 | |
| 122 | def load(self, input_context: Optional[tf.distribute.InputContext] = None): |
| 123 | """Returns a tf.dataset.Dataset.""" |
| 124 | reader = input_reader.InputReader( |
| 125 | dataset_fn=dataset_fn.pick_dataset_fn(self._params.file_type), |
| 126 | params=self._params, |
| 127 | decoder_fn=self._decode, |
| 128 | parser_fn=self._parse) |
| 129 | return reader.read(input_context) |
no test coverage detected