(self, dataset)
| 607 | self._create_iterator(dataset) |
| 608 | |
| 609 | def _create_iterator(self, dataset): |
| 610 | # pylint: disable=protected-access |
| 611 | dataset = dataset._apply_options() |
| 612 | ds_variant = dataset._variant_tensor |
| 613 | self._element_spec = dataset.element_spec |
| 614 | self._flat_output_types = structure.get_flat_tensor_types( |
| 615 | self._element_spec) |
| 616 | self._flat_output_shapes = structure.get_flat_tensor_shapes( |
| 617 | self._element_spec) |
| 618 | with ops.device(ds_variant.device): |
| 619 | self._iterator_resource, self._deleter = ( |
| 620 | gen_dataset_ops.anonymous_iterator_v2( |
| 621 | output_types=self._flat_output_types, |
| 622 | output_shapes=self._flat_output_shapes)) |
| 623 | gen_dataset_ops.make_iterator(ds_variant, self._iterator_resource) |
| 624 | # Delete the resource when this object is deleted |
| 625 | self._resource_deleter = IteratorResourceDeleter( |
| 626 | handle=self._iterator_resource, |
| 627 | device=self._device, |
| 628 | deleter=self._deleter) |
| 629 | |
| 630 | def __iter__(self): |
| 631 | return self |
no test coverage detected