Searched refs:maybe_squeeze_weights (Results 1 – 2 of 2) sorted by relevance
129 maybe_squeeze_weights = lambda: array_ops.squeeze(sample_weight, [-1]) function139 math_ops.equal(rank_diff, 1), maybe_squeeze_weights,
147 maybe_squeeze_weights = lambda: weights function149 maybe_squeeze_weights = lambda: array_ops.squeeze(weights, [-1]) function153 math_ops.equal(rank_diff, 1), maybe_squeeze_weights,