(self, batch_size)
| 2616 | return DatasetSpec(value.element_spec) # pylint: disable=protected-access |
| 2617 | |
| 2618 | def _batch(self, batch_size): |
| 2619 | return DatasetSpec( |
| 2620 | self._element_spec, |
| 2621 | tensor_shape.TensorShape([batch_size]).concatenate(self._dataset_shape)) |
| 2622 | |
| 2623 | def _unbatch(self): |
| 2624 | if self._dataset_shape.ndims == 0: |