diff options
author | Jiri Simsa <jsimsa@google.com> | 2018-09-05 15:07:07 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-05 15:20:08 -0700 |
commit | 7e2577b0984a1d8f41af97942fcdf5b9f1ff8622 (patch) | |
tree | e197a9ec41a64911fbe7ef306926b4124024d73f /tensorflow/contrib/data | |
parent | 7288b3da07e7c88ac0996fd5ddb830e501c8f114 (diff) |
[tf.data] Minor fix to remove unnecessary difference between the implementations of the batch and padded batch reducers.
PiperOrigin-RevId: 211706766
Diffstat (limited to 'tensorflow/contrib/data')
-rw-r--r-- | tensorflow/contrib/data/python/ops/batching.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/data/python/ops/batching.py b/tensorflow/contrib/data/python/ops/batching.py index 9c2001c34f..367c159dc5 100644 --- a/tensorflow/contrib/data/python/ops/batching.py +++ b/tensorflow/contrib/data/python/ops/batching.py @@ -272,9 +272,9 @@ def _padded_batch_dense_window(dataset, padded_shape, padding_value=None): padding_value = 0 def batch_init_fn(_): - return array_ops.fill( - array_ops.concat([np.array([0], dtype=np.int32), padded_shape], 0), - constant_op.constant(padding_value, dtype=dataset.output_types)) + batch_shape = array_ops.concat( + [np.array([0], dtype=np.int32), padded_shape], 0) + return gen_array_ops.empty(batch_shape, dtype=dataset.output_types) def batch_reduce_fn(state, value): return array_ops.concat([state, [value]], 0) |