From d89b24ff13a7b95da0cd976b47863974c6b9dec3 Mon Sep 17 00:00:00 2001 From: James Qin Date: Tue, 28 Apr 2020 14:31:23 -0700 Subject: [PATCH] Correct typo and better doc. PiperOrigin-RevId: 308898378 Change-Id: I0ec7d91a7398b0470ca8ded6d9b2609ca737c8b2 --- tensorflow/python/tpu/tpu_feed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/tpu/tpu_feed.py b/tensorflow/python/tpu/tpu_feed.py index 2b01eeb3934..ce5f9aa6b8b 100644 --- a/tensorflow/python/tpu/tpu_feed.py +++ b/tensorflow/python/tpu/tpu_feed.py @@ -556,8 +556,8 @@ class InfeedQueue(object): """Generates the host-side Ops to enqueue the shards of a tuple. sharded_inputs is a list, one for each shard, of lists of - Tensors. sharded_inputs[0] is the tuple of Tensors to use to feed - shard 0 if the queue. Returns the host-side Ops that must be run to + Tensors. sharded_inputs[i] is the tuple of Tensors to use to feed + shard i of the queue. Returns the host-side Ops that must be run to enqueue the sharded tuple. The Op for shard i is colocated with the inputs for shard i.