Searched refs:_flatten_outer_dims (Results 1 – 1 of 1) sorted by relevance
2767 def _flatten_outer_dims(logits): function3140 precise_logits = _flatten_outer_dims(precise_logits)3141 labels = _flatten_outer_dims(labels)