(self, shard=0, num_shards=1)
| 145 | return feats |
| 146 | |
| 147 | def as_dataset(self, shard=0, num_shards=1): |
| 148 | works = tf.data.Dataset.from_tensor_slices(self._partition(shard, num_shards)) |
| 149 | work_reader = tf.data.experimental.parallel_interleave( |
| 150 | self._make_dataset, |
| 151 | cycle_length=self.reader_threads, |
| 152 | block_length=self.reader_buffer, |
| 153 | sloppy=True) |
| 154 | ds = works.apply(work_reader) |
| 155 | return ds.map(self._next, self.parser_threads).prefetch(self.parser_buffer) |
no test coverage detected