Searched refs:expand_weights (Results 1 – 2 of 2) sorted by relevance
106 expand_weights = lambda: array_ops.expand_dims(sample_weight, [-1]) function108 math_ops.equal(rank_diff, -1), expand_weights, lambda: sample_weight)
218 expand_weights = lambda: array_ops.expand_dims(sample_weight, [-1]) function220 math_ops.equal(rank_diff, -1), expand_weights, lambda: sample_weight)