Searched refs:margin_diff (Results 1 – 1 of 1) sorted by relevance
184 auto margin_diff = (margin - self); in hinge_embedding_loss() local189 auto margin_clamp = (margin_diff._fw_grad(/*level*/ 0).defined() && in hinge_embedding_loss()190 isTensorSubclassLike(margin_diff)) in hinge_embedding_loss()191 ? margin_diff.clamp_min(0) in hinge_embedding_loss()192 : margin_diff.clamp_min_(0); in hinge_embedding_loss()