diff options
| -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. |