Searched refs:maybe_squeeze_dims (Results 1 – 2 of 2) sorted by relevance
79 maybe_squeeze_dims = lambda: control_flow_ops.cond( # pylint: disable=g-long-lambda function82 math_ops.equal(1, rank_diff), maybe_squeeze_dims, squeeze_dims)
191 maybe_squeeze_dims = lambda: control_flow_ops.cond( # pylint: disable=g-long-lambda function194 math_ops.equal(1, rank_diff), maybe_squeeze_dims, squeeze_dims)