Searched defs:log_weight (Results 1 – 2 of 2) sorted by relevance
98 auto log_weight = (pos_weight - 1).mul(target).add_(1); in binary_cross_entropy_with_logits_hack() local
372 auto log_weight = (pos_weight - 1).mul(target).add_(1); in binary_cross_entropy_with_logits() local