Searched refs:_MultiHead (Results 1 – 1 of 1) sorted by relevance
496 return _MultiHead(heads, loss_merger=_weighted_loss_merger)1529 class _MultiHead(Head): class