From 49eac7f8e21d3dcfa1424a83fa8768c33d94b4aa Mon Sep 17 00:00:00 2001 From: Edward Loper Date: Tue, 10 Mar 2020 10:10:31 -0700 Subject: [PATCH] Minor optimization for RowPartition.from_uniform_row_length PiperOrigin-RevId: 300118346 Change-Id: I9b653a73ee65668fea34931c888052d44520629c --- tensorflow/python/ops/ragged/row_partition.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tensorflow/python/ops/ragged/row_partition.py b/tensorflow/python/ops/ragged/row_partition.py index db31af15455..ae87c824d0d 100644 --- a/tensorflow/python/ops/ragged/row_partition.py +++ b/tensorflow/python/ops/ragged/row_partition.py @@ -519,10 +519,9 @@ class RowPartition(object): if nrows is None: if const_row_length is None: # Avoid division by zero if uniform_row_length==0 (and nvals==0). - rowlen_or_1 = control_flow_ops.cond( - math_ops.equal(uniform_row_length, 0), - lambda: constant_op.constant(1, uniform_row_length.dtype), - lambda: uniform_row_length) + rowlen_or_1 = math_ops.maximum( + uniform_row_length, + constant_op.constant(1, uniform_row_length.dtype)) nrows = nvals // rowlen_or_1 elif const_row_length == 0: nrows = 0