diff options
author | 2018-04-12 12:39:48 -0700 | |
---|---|---|
committer | 2018-04-12 12:42:35 -0700 | |
commit | f95906527e92a151a424b60a109d2361e20d610b (patch) | |
tree | 59effd686a2ccc4e80f1c39c4ace895e4e887825 | |
parent | 1a014c6d62bad0e58e3c8a1e31beb396daa19c13 (diff) |
Fix comment of bucket_by_sequence_length about return type of
element_length_func. Current code requires tf.int32 in order to compare with
buckets_min which is int32.
PiperOrigin-RevId: 192652917
-rw-r--r-- | tensorflow/contrib/data/python/ops/grouping.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/data/python/ops/grouping.py b/tensorflow/contrib/data/python/ops/grouping.py index 36591c055a..0531f9cbb9 100644 --- a/tensorflow/contrib/data/python/ops/grouping.py +++ b/tensorflow/contrib/data/python/ops/grouping.py @@ -108,7 +108,7 @@ def bucket_by_sequence_length(element_length_func, fraction of padding in a batch which increases training step efficiency. Args: - element_length_func: function from element in `Dataset` to `tf.int64`, + element_length_func: function from element in `Dataset` to `tf.int32`, determines the length of the element, which will determine the bucket it goes into. bucket_boundaries: `list<int>`, upper length boundaries of the buckets. |