Searched refs:_flatten_outer_dims (Results 1 – 1 of 1) sorted by relevance
3550 def _flatten_outer_dims(logits): function3959 precise_logits = _flatten_outer_dims(precise_logits)3960 labels = _flatten_outer_dims(labels)