Fix make_initializable_iterator comment

PiperOrigin-RevId: 227598505
This commit is contained in:
A. Unique TensorFlower 2019-01-02 16:08:42 -08:00 committed by TensorFlower Gardener
parent f0dde544f2
commit 0b6177c2fa

View File

@ -1756,8 +1756,8 @@ def make_initializable_iterator(dataset):
RuntimeError: If eager execution is enabled. RuntimeError: If eager execution is enabled.
""" """
try: try:
# Call the defined `make_one_shot_iterator()` if there is one, because some # Call the defined `make_initializable_iterator()` if there is one, because
# datasets (e.g. for prefetching) override its behavior. # some datasets (e.g. for prefetching) override its behavior.
return dataset.make_initializable_iterator() return dataset.make_initializable_iterator()
except AttributeError: except AttributeError:
return DatasetV1Adapter(dataset).make_initializable_iterator() return DatasetV1Adapter(dataset).make_initializable_iterator()