diff options
author | 2017-06-09 11:57:46 -0700 | |
---|---|---|
committer | 2017-06-09 12:01:35 -0700 | |
commit | 1700ac827237992143144a5763a72d56b2da7127 (patch) | |
tree | d56ec755666fe98c2791337259e86b791f31f814 | |
parent | fd584b5e9daf57c827b0ee652317ab0f205ae4bb (diff) |
Use correct module when calling reduce_prod.
PiperOrigin-RevId: 158544698
-rw-r--r-- | tensorflow/python/ops/distributions/util.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/ops/distributions/util.py b/tensorflow/python/ops/distributions/util.py index 74a7b53a3c..05c6f4da57 100644 --- a/tensorflow/python/ops/distributions/util.py +++ b/tensorflow/python/ops/distributions/util.py @@ -544,7 +544,7 @@ def fill_lower_triangular(x, validate_args=False, name="fill_lower_triangular"): m = np.prod(batch_shape).astype(np.int32) else: batch_shape = array_ops.shape(x)[:-1] - m = array_ops.reduce_prod(array_ops.shape(x)[:-1]) + m = math_ops.reduce_prod(array_ops.shape(x)[:-1]) batch_ids = math_ops.range(m) # Assemble the tril_ids into batch,tril_id pairs. |