Method
__init__
(self,
number_of_tuple_elements,
device_assignment,
host_id,
input_partition_dims=None,
tuple_types=None,
tuple_shapes=None,
name=None)
Source from the content-addressed store, hash-verified
| 747 | """ |
| 748 | |
| 749 | def __init__(self, |
| 750 | number_of_tuple_elements, |
| 751 | device_assignment, |
| 752 | host_id, |
| 753 | input_partition_dims=None, |
| 754 | tuple_types=None, |
| 755 | tuple_shapes=None, |
| 756 | name=None): |
| 757 | super(_PartitionedInfeedQueue, self).__init__( |
| 758 | number_of_tuple_elements=number_of_tuple_elements, |
| 759 | tuple_types=tuple_types, |
| 760 | tuple_shapes=None, |
| 761 | shard_dimensions=None, |
| 762 | name="PartitionedInfeedQueue" if name is None else name) |
| 763 | self._input_partition_dims = input_partition_dims |
| 764 | self._host_id = host_id |
| 765 | self._device_assignment = device_assignment |
| 766 | |
| 767 | def generate_dequeue_op(self, tpu_device=0): |
| 768 | """Generate TPU dequeue ops. |
Callers
nothing calls this directly
Tested by
no test coverage detected