Searched refs:_LossOnlyHead (Results 1 – 1 of 1) sorted by relevance
463 return _LossOnlyHead(loss_fn, head_name=head_name)1451 class _LossOnlyHead(Head): class1648 if isinstance(head, _LossOnlyHead):1738 if isinstance(head, _LossOnlyHead):