Searched refs:_move_dim_to_end (Results 1 – 1 of 1) sorted by relevance
3126 def _move_dim_to_end(tensor, dim_index, rank): function3134 precise_logits = _move_dim_to_end(precise_logits, axis, input_rank)3135 labels = _move_dim_to_end(labels, axis, input_rank)